RTIInternational / gobbli

Deep learning with text doesn't have to be scary.
Apache License 2.0
275 stars 23 forks source link

SKLearnClassifier interface to scikit-learn models #16

Closed jasonnance closed 4 years ago

jasonnance commented 4 years ago

Description of Changes

Implements a flexible SKLearnClassifier gobbli model which can wrap any scikit-learn estimator that accepts text and outputs predicted label probabilities. Adds some helper functions for simple default TF-IDF vectorized logistic regression.