jam/test/test-dt1.js

20 lines
560 B
JavaScript
Raw Normal View History

2024-08-27 00:15:25 +02:00
var x = [[0, 0, 0], [0, 1, 1], [1, 1, 0], [2, 2, 2], [1, 2, 2], [2, 1, 2]];
var y = ['A', 'A', 'B', 'B', 'C', 'C'];
var model = ml.learn({
algorithm:ml.ML.DT,
x:x,
y:y,
eps:0.2
});
print(toJSON(model).length+' Bytes')
var test_data =[[0, 1.2, 0],
[2.1, 2, 3],
[2.1,1.1,2.0]];
print(ml.print(model))
print(ml.classify(model,x))
print(ml.classify(model,x.map(function (row) { return row.map(function (col) { return col+random(-0.3,0.3,0.001) })})))
print(ml.classify(model,test_data))
print(ml.classify(model,[1,2,3]))