Classifier API

Classifiers

class small_text.classifiers.classification.Classifier

Abstract base class for classifiers that can be used with the active learning components.

class small_text.classifiers.classification.SklearnClassifier(model, num_classes, multi_label=False)

An adapter for using scikit-learn estimators.

Notes

The multi-label settings currently assumes that the underlying classifer returns a sparse matrix if trained on sparse data.

Factories

class small_text.classifiers.factories.SklearnClassifierFactory(base_estimator, num_classes, kwargs={})