/
bayes.test.js
36 lines (32 loc) · 1.44 KB
/
bayes.test.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
var assert = require('assert');
var ss = require('../');
describe('bayes', function() {
it('makes an easy call with one training round', function() {
var bayes = ss.bayesian();
bayes.train({ species: 'Cat' }, 'animal');
assert.deepEqual(bayes.score({ species: 'Cat' }), { animal: 1 });
});
it('makes fify-fifty call', function() {
var bayes = ss.bayesian();
bayes.train({ species: 'Cat' }, 'animal');
bayes.train({ species: 'Cat' }, 'chair');
assert.deepEqual(bayes.score({ species: 'Cat' }), { animal: 0.5, chair: 0.5 });
});
it('makes seventy-five/twenty-five call', function() {
var bayes = ss.bayesian();
bayes.train({ species: 'Cat' }, 'animal');
bayes.train({ species: 'Cat' }, 'animal');
bayes.train({ species: 'Cat' }, 'animal');
bayes.train({ species: 'Cat' }, 'chair');
assert.deepEqual(bayes.score({ species: 'Cat' }), { animal: 0.75, chair: 0.25 });
});
it('classifies multiple things', function() {
var bayes = ss.bayesian();
bayes.train({ species: 'Cat' }, 'animal');
bayes.train({ species: 'Dog' }, 'animal');
bayes.train({ species: 'Dog' }, 'animal');
bayes.train({ species: 'Cat' }, 'chair');
assert.deepEqual(bayes.score({ species: 'Cat' }), { animal: 0.25, chair: 0.25 });
assert.deepEqual(bayes.score({ species: 'Dog' }), { animal: 0.5, chair: 0 });
});
});