hackarada / bayes-classifier

Naive Bayes Classifier in JavaScript - Node

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

http://lab.moogs.io/bayes-classifier

Install

npm install bayes-classifier

bower install bayes-classifier

Usage

var BayesClassifier = require('bayes-classifier');
var classifier = new BayesClassifier();

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.'
];

classifier.addDocuments(positiveDocuments, 'positive');
classifier.addDocuments(negativeDocuments, 'negative');

classifier.train();

console.log(classifier.classify('I heard the mexican restaurant is great!')); // "positive"
console.log(classifier.classify('I don\'t want to eat there again.')); // "negative"
console.log(classifier.classify('The torta is epicly bad.')); // "negative"
console.log(classifier.classify('The torta is horribly awesome.')); // "positive"

console.log(classifier.getClassifications('Burritos are the meaning of life.'));
/*
 [ { label: 'positive', value: 0.22222222222222224 },
   { label: 'negative', value: 0.11111111111111112 } ]
*/

API

classifier.addDocument(doc, class)

classifier.addDocuments([docs], class)

classifier.train()

classifier.classify(doc)

classifier.getClassifications(doc)

License

MIT