115 lines
2.6 KiB
JavaScript
115 lines
2.6 KiB
JavaScript
var ss = require('../');
|
|
var test = require('tape');
|
|
|
|
test('bayes', function(t) {
|
|
test('makes an easy call with one training round', function(t) {
|
|
var bayes = ss.bayesian();
|
|
bayes.train({
|
|
species: 'Cat'
|
|
}, 'animal');
|
|
t.deepEqual(bayes.score({
|
|
species: 'Cat'
|
|
}), {
|
|
animal: 1
|
|
});
|
|
t.end();
|
|
});
|
|
|
|
test('makes fify-fifty call', function(t) {
|
|
var bayes = ss.bayesian();
|
|
bayes.train({
|
|
species: 'Cat'
|
|
}, 'animal');
|
|
bayes.train({
|
|
species: 'Cat'
|
|
}, 'chair');
|
|
t.deepEqual(bayes.score({
|
|
species: 'Cat'
|
|
}), {
|
|
animal: 0.5,
|
|
chair: 0.5
|
|
});
|
|
t.end();
|
|
});
|
|
|
|
test('makes seventy-five/twenty-five call', function(t) {
|
|
var bayes = ss.bayesian();
|
|
bayes.train({
|
|
species: 'Cat'
|
|
}, 'animal');
|
|
bayes.train({
|
|
species: 'Cat'
|
|
}, 'animal');
|
|
bayes.train({
|
|
species: 'Cat'
|
|
}, 'animal');
|
|
bayes.train({
|
|
species: 'Cat'
|
|
}, 'chair');
|
|
t.deepEqual(bayes.score({
|
|
species: 'Cat'
|
|
}), {
|
|
animal: 0.75,
|
|
chair: 0.25
|
|
});
|
|
t.end();
|
|
});
|
|
|
|
test('tests multiple properties', function(t) {
|
|
var bayes = ss.bayesian();
|
|
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.deepEqual(bayes.score({
|
|
color: 'white'
|
|
}), {
|
|
animal: 0,
|
|
chair: 0.2
|
|
});
|
|
t.end();
|
|
});
|
|
|
|
test('classifies multiple things', function(t) {
|
|
var bayes = ss.bayesian();
|
|
bayes.train({
|
|
species: 'Cat'
|
|
}, 'animal');
|
|
bayes.train({
|
|
species: 'Dog'
|
|
}, 'animal');
|
|
bayes.train({
|
|
species: 'Dog'
|
|
}, 'animal');
|
|
bayes.train({
|
|
species: 'Cat'
|
|
}, 'chair');
|
|
t.deepEqual(bayes.score({
|
|
species: 'Cat'
|
|
}), {
|
|
animal: 0.25,
|
|
chair: 0.25
|
|
});
|
|
t.deepEqual(bayes.score({
|
|
species: 'Dog'
|
|
}), {
|
|
animal: 0.5,
|
|
chair: 0
|
|
});
|
|
t.end();
|
|
});
|
|
t.end();
|
|
});
|