Naive Bayes classifier
This is a Naive Bayes classifier implementation written in JavaScript.
I took out the relevant algorithms from the appratus and natural modules, and also the Porter stemmer algorithm. All credit goes to them.
Demo
https://lab.miguelmota.com/bayes-classifier
Install
npm install bayes-classifier
bower install bayes-classifier
Usage
var BayesClassifier = var classifier = var positiveDocuments = `I love tacos.` `Dude, that burrito was epic!` `Holy cow, these nachos are so good and tasty.` `I am drooling over the awesome bean and cheese quesadillas.` var negativeDocuments = `Gross, worst taco ever.` `The buritos gave me horrible diarrhea.` `I'm going to puke if I eat another bad nacho.` `I'd rather die than eat those nasty enchiladas.` classifierclassifier classifier console // "positive"console // "negative"console // "negative"console // "positive" console/* [ { label: 'positive', value: 0.22222222222222224 }, { label: 'negative', value: 0.11111111111111112 } ]*/
Restoring a classifier to avoid re-training data
// Storing classifiervar storeFile = `/store.json`fs // ... // Restoring classifiervar classifier = var storedClassifier = classifier
API
classifier.addDocument(doc, class)
classifier.addDocuments([docs], class)
classifier.train()
classifier.classify(doc)
classifier.getClassifications(doc)
classifier.restore(classifier)
Test
npm test
License
MIT