choco-bot/brain.js

Whatsapp-bot

const natural = require('natural');
const classifier = new natural.BayesClassifier();
const trainingData = require('./trainingData.json');

trainingData.forEach(item => {
    classifier.addDocument(item.text, item.intent);
});

classifier.train();

module.exports = {
    classifyIntent: (message) => {
        const classified = classifier.getClassifications(message).sort((a, b) => b.value - a.value);
        const intent = classified[0].label;
        const response = trainingData.find(data => data.intent === intent).response;
        return { intent, response };
    },
};