sbosse/jam
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Tue 27 Aug 00:14:56 CEST 2024

Browse Source
This commit is contained in:
sbosse 2024-08-27 00:15:30 +02:00
parent f2464e6e16
commit 5b883273cd
1 changed files with 31 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
test/test-id3.js Normal file
View File

@ -0,0 +1,31 @@
var training_data = [
{"color":"blue", "shape":"square", "liked":false},
{"color":"red", "shape":"square", "liked":false},
{"color":"blue", "shape":"circle", "liked":true},
{"color":"red", "shape":"circle", "liked":true},
{"color":"blue", "shape":"hexagon", "liked":false},
{"color":"red", "shape":"hexagon", "liked":false},
{"color":"yellow", "shape":"hexagon", "liked":true},
{"color":"yellow", "shape":"circle", "liked":true}
]
var test_data = [
{"color":"blue", "shape":"hexagon", "liked":false},
{"color":"red", "shape":"hexagon", "liked":false},
{"color":"yellow", "shape":"hexagon", "liked":true},
{"color":"yellow", "shape":"circle", "liked":true}
];
var target = "liked";
var features = ["color", "shape"];
var model = ml.learn({
algorithm:ml.ML.ID3,
data:training_data,
target:target,
features:features
})
print(ml.print(model))
print(toJSON(model).length+' Bytes')
print(ml.classify(model,test_data))