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
37
38
39
40
41
42
43
44
45
|
use strict;
use Test;
BEGIN { plan tests => 13 };
use Algorithm::NaiveBayes;
ok(1); # If we made it this far, we're loaded.
my $nb = Algorithm::NaiveBayes->new(purge => 0);
ok $nb;
# Populate
$nb->add_instance( attributes => _hash(qw(sheep very valuable farming)),
label => 'farming' );
ok $nb->labels, 1;
$nb->add_instance( attributes => _hash(qw(farming requires many kinds animals)),
label => ['farming'] );
ok $nb->labels, 1;
$nb->add_instance( attributes => _hash(qw(vampires drink blood vampires may staked)),
label => ['vampire'] );
ok $nb->labels, 2;
$nb->add_instance( attributes => _hash(qw(vampires cannot see their images mirrors)),
label => ['vampire'] );
ok $nb->labels, 2;
# Train
$nb->train;
ok $nb->purge, 0;
# Predict
my $h = $nb->predict( attributes => _hash(qw(i would like to begin farming sheep)) );
ok $h;
ok $h->{farming} > 0.5;
ok $h->{vampire} < 0.5;
$h = $nb->predict( attributes => _hash(qw(i see that many vampires may have eaten my beautiful daughter's blood)) );
ok $h;
ok $h->{farming} < 0.5;
ok $h->{vampire} > 0.5;
################################################################
sub _hash { +{ map {$_,1} @_ } }
|
