A Scikit-learn wrapper for the NBSVM algorithm. To read more about the same, go here
Use the package manager pip to install nbsvm-sklearn.
pip install nbsvm-sklearnCode has been tested on python>=3.6
from nbsvm import NBSVMClassifier
clf = NBSVMClassifier() # initialize the model
clf.fit(X, y) # train the classifier; y{0,1}
clf.predict(X) # get binary predictionsFull Code documentation available here
Version 0.0.5: Added Platt-scaling
- Add support for multi-class classification
- Handle sparse matrices as inputs
- Handle
pd.Seriesinput format for labels
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.