Tue 27 Aug 00:14:56 CEST 2024

test/test-rf2.js

@@ -0,0 +1,34 @@
+var data = load('test-data-iris.json')
+
+var datac = ml.preprocess(data,'xmy',{features:['length','width','petal_length','petal_width'],
+                                      target:'species'});
+// print(datac)
+var labels = [
+  'setosa',
+  'versicolor',
+  'virginica'
+]
+var x = datac.x;
+var y = datac.y;
+
+var t0=time()
+var model = ml.learn({
+    algorithm:ml.ML.RF,
+    labels:labels,
+    numTrees:5,
+    maxDepth:10,
+    numTries:10,
+    weakType:false,
+    x:x, 
+    y:y,
+  });
+  
+var t1=time()
+var result = ml.classify(model,x).map(ml.best).map(function (r,i) { r.y=y[i]; return r } )
+var t2=time()
+print(result)
+var correct=0,wrong=0;
+result.forEach(function (r) { if (r.value==r.y) correct++; else wrong++ });
+print('Correct='+correct+', wrong='+wrong);
+print(toJSON(model).length)
+print(t1-t0,t2-t1)