deno.land / x / simplestatistics@v7.8.3 / test / bayes.test.js
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184/* eslint no-shadow: 0 */
const BayesianClassifier = require("../dist/simple-statistics.js").BayesianClassifier;const test = require("tap").test;
test("BayesianClassifier", function (t) { t.test("makes an easy call with one training round", function (t) { const bayes = new BayesianClassifier(); bayes.train( { species: "Cat" }, "animal" ); t.same( bayes.score({ species: "Cat" }), { animal: 1 } ); t.end(); });
t.test("makes fify-fifty call", function (t) { const bayes = new BayesianClassifier(); bayes.train( { species: "Cat" }, "animal" ); bayes.train( { species: "Cat" }, "chair" ); t.same( bayes.score({ species: "Cat" }), { animal: 0.5, chair: 0.5 } ); t.end(); });
t.test("makes seventy-five/twenty-five call", function (t) { const bayes = new BayesianClassifier(); bayes.train( { species: "Cat" }, "animal" ); bayes.train( { species: "Cat" }, "animal" ); bayes.train( { species: "Cat" }, "animal" ); bayes.train( { species: "Cat" }, "chair" ); t.same( bayes.score({ species: "Cat" }), { animal: 0.75, chair: 0.25 } ); t.end(); });
t.test("tests multiple properties", function (t) { const bayes = new BayesianClassifier(); bayes.train( { species: "Cat" }, "animal" ); bayes.train( { species: "Cat" }, "animal" ); bayes.train( { species: "Cat" }, "animal" ); bayes.train( { species: "Cat" }, "chair" ); bayes.train( { species: "Cat", color: "white" }, "chair" ); t.same( bayes.score({ color: "white" }), { animal: 0, chair: 0.2 } ); t.end(); });
t.test("classifies multiple things", function (t) { const bayes = new BayesianClassifier(); bayes.train( { species: "Cat" }, "animal" ); bayes.train( { species: "Dog" }, "animal" ); bayes.train( { species: "Dog" }, "animal" ); bayes.train( { species: "Cat" }, "chair" ); t.same( bayes.score({ species: "Cat" }), { animal: 0.25, chair: 0.25 } ); t.same( bayes.score({ species: "Dog" }), { animal: 0.5, chair: 0 } ); t.end(); }); t.end();});
Version Info