Tue 27 Aug 00:14:56 CEST 2024

This commit is contained in:
sbosse 2024-08-27 00:15:00 +02:00
parent 4bb33be0ff
commit cecb2f3085

24
test/test-dt2.js Normal file
View File

@ -0,0 +1,24 @@
var data = [
{a:0, b:0, c:0, y:'A'},
{a:0, b:1, c:1, y:'A'},
{a:1, b:1, c:0, y:'B'},
{a:2, b:2, c:2, y:'B'},
{a:1, b:2, c:2, y:'C'},
{a:2, b:1, c:2, y:'C'}
];
var model = ml.learn({
algorithm:ml.ML.DT,
data:data,
target:'y',
features:['a','b','c'],
eps:0.2
});
print(toJSON(model).length+' Bytes')
var test_data =[{a:0, b:1.2, c:0},
{a:2.1, b:2, c:3},
{a:2.1, b:1.1, c:2.0}];
print(ml.print(model))
print(ml.classify(model,test_data))
print(ml.classify(model,{a:1,b:2,c:3}))