Skip to content
/ nbsvm Public

Multiclass Naive Bayes SVM (NB-SVM) - Learns a multiclass classifier (OneVsRest) based on word ngrams.

Notifications You must be signed in to change notification settings

lrei/nbsvm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

Multiclass Naive Bayes SVM (NB-SVM)

Luis Rei luis.rei@ijs.si @lmrei http://luisrei.com

Learns a multiclass classifier (OneVsRest) based on word ngrams.

Uses scikit learn. Reads input from TSV files. Text is expected to already be tokenized.

Original paper (binary classifier): Sida Wang and Christopher D. Manning: Baselines and Bigrams: Simple, Good Sentiment and Topic Classification; ACL 2012. http://nlp.stanford.edu/pubs/sidaw12_simple_sentiment.pdf

Based on a work at https://github.com/mesnilgr/nbsvm: Naive Bayes SVM by Grégoire Mesnil

nbsvm2.py vs nbsvm.py

The second version is an initial attempt to use numpy/scipy/sklearn more.

Warranty

No guarantee that this works or is correctly implemented.

Notes:

  • I modified the code from Mesnil for multiclass because I had neutral sentiment.
  • This does not use the MNB + SVM ensemble from the original paper, just the SVM part (same as Mesnil)

License

Creative Commons License
Multiclass Naive Bayes SVM (NB-SVM) by Luis Rei is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
Based on a work at https://github.com/mesnilgr/nbsvm.

About

Multiclass Naive Bayes SVM (NB-SVM) - Learns a multiclass classifier (OneVsRest) based on word ngrams.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages