jam/test/test-rf1.js

18 lines
387 B
JavaScript

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 y = [1, 1, 2, 2, 3, 3];
print(x)
print(y)
var model = ml.learn({
algorithm:ml.ML.RF,
x:x,
y:y,
});
var test_data =[[0, 1.2, 0],
[2.1, 2, 3],
[2.1,1.1,2.0]];
// print(inspect(model,_,10))
print(ml.classify(model,test_data))