nbsvm-sklearn

A Scikit-learn wrapper for the NBSVM algorithm. To read more about the same, go here

Installation

Use the package manager pip to install nbsvm-sklearn.

pip install nbsvm-sklearn

Code has been tested on python>=3.6

Usage

from nbsvm import NBSVMClassifier

clf = NBSVMClassifier() # initialize the model
clf.fit(X, y) # train the classifier; y{0,1}
clf.predict(X) # get binary predictions

Full Code documentation available here

Updates

Version 0.0.5: Added Platt-scaling

Future Work

• Add support for multi-class classification
• Handle sparse matrices as inputs
• Handle pd.Series input format for labels

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

Please make sure to update tests as appropriate.

License

MIT