mirror of
https://github.com/gosticks/DefinitelyTyped.git
synced 2025-10-16 12:05:41 +00:00
34 lines
1.1 KiB
TypeScript
34 lines
1.1 KiB
TypeScript
import BayesClassifier = require('bayes-classifier');
|
|
|
|
const classifier = new BayesClassifier();
|
|
|
|
const 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.`,
|
|
];
|
|
|
|
const 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();
|
|
|
|
classifier.classify(`I heard the mexican restaurant is great!`); // "positive"
|
|
classifier.classify(`I don't want to eat there again.`); // "negative"
|
|
classifier.classify(`The torta is epicly bad.`); // "negative"
|
|
classifier.classify(`The torta is tasty.`); // "positive"
|
|
|
|
classifier.getClassifications(`Burritos are the meaning of life.`);
|
|
/*
|
|
[ { label: 'positive', value: 0.22222222222222224 },
|
|
{ label: 'negative', value: 0.11111111111111112 } ]
|
|
*/
|