nessie.models.text.lightgbm_text_classifier

Module Contents

Classes

LgbmTextClassifier

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.