nessie.models.text.lightgbm_text_classifier
Module Contents
Classes
This model uses a sentence embedder like S-BERT to embed sentences, these are inputs to |
- class nessie.models.text.lightgbm_text_classifier.LgbmTextClassifier(embedder: nessie.models.featurizer.SentenceEmbedder)
Bases:
nessie.models.text.sklean_text_classifier.SklearnTextClassifier
This model uses a sentence embedder like S-BERT to embed sentences, these are inputs to train a model with scikit learn API.