text-classifier v1.1.1

Text Classifier

Example

'use strict';

const TextClassifier = require('text-classifier');

let classifier = new TextClassifier;

// or
let classifier = new TextClassifier({
    // replacers for prepareString function
    replacers: [
        [/[\-_]+/g, ' '],
        [/[^\w\d\s]+/g, ''],
        // ...
    ]
});

// Learn #1
// positive
classifier.learn('amazing, awesome movie!! Yeah!! Oh boy.', 'positive');
classifier.learn('Sweet, this is incredibly, amazing, perfect, great!!', 'positive');
// negative
classifier.learn('terrible, shitty thing. Damn.', 'negative');

// Learn #2
// positive
classifier.learn([
    'amazing, awesome movie!! Yeah!! Oh boy.',
    'Sweet, this is incredibly, amazing, perfect, great!!'
], 'positive');
// negative
classifier.learn('terrible, shitty thing. Damn.', 'negative');

// Learn #3
classifier.learn({
    // positive
    'amazing, awesome movie!! Yeah!! Oh boy.': 'positive',
    'Sweet, this is incredibly, amazing, perfect, great!!': 'positive',
    // negative
    'terrible, shitty thing. Damn.': 'negative'
});

// Classify
classifier.classify('awesome, cool, amazing!! Yay.');
//=> { score: 0.6153846153846154, textClass: 'positive' }

// To JSON
classifier.toJSON();
//=> JSON string

// Load from JSON or Object
classifier.load(jsonOrObject);