level-naive-bayes
v1.0.0
Published
Bayes text classifier that runs on top of leveldb
Downloads
3
Readme
level-naive-bayes
Naive Bayes text classifier that runs on top of leveldb so you can work with datasets that doesn't fit in memory. Based on the bayes module.
npm install level-naive-bayes
Usage
var bayes = require('level-naive-bayes')
var nb = bayes(db) // where db is a levelup instance
nb.train('positive', 'amazing, awesome movie!! Yeah!! Oh boy.', function() {
nb.train('positive', 'this is incredibly, amazing, perfect, great!', function() {
nb.train('negative', 'terrible, shitty thing. Damn. Sucks!!', function() {
nb.classify('awesome, cool, amazing!! Yay.', function(err, category) {
console.log('category is '+category)
})
})
})
})
API
nb = bayes(db, [options])
Creates a new instance. db
should be a levelup.
Options include:
{
tokenize: function(str) {
return str.split(' ') // pass in custom tokenizer
}
}
nb.train(category, text, cb)
Train the classifier with the given text for a category. If the text is already tokenized pass in an array of tokens instead of text
nb.classify(text, cb)
Classify the given text into a category. If the text is already tokenized pass in an array of tokens instead of text
License
MIT