Merge pull request #1 from thunterdb/1606-trees

adding experiments to the yaml file
2016-07-01 11:46:41 -07:00 · 2016-07-01 11:46:41 -07:00 · c2f0a35db4
commit c2f0a35db4
parent c15d083fe7 813bd8ad59
1 changed files with 22 additions and 0 deletions
--- a/src/main/scala/configs/mllib-small.yaml
+++ b/src/main/scala/configs/mllib-small.yaml
@ -33,3 +33,25 @@ benchmarks:
      tol: 0.0
      maxIter: 10
      regParam: 0.1
+  - name: classification.DecisionTreeClassification
+    params:
+      numExamples: 100
+      numTestExamples: 10
+      depth: 3
+      numClasses: 4
+      numFeatures: 5
+  - name: classification.RandomForestClassification
+    params:
+      numExamples: 100
+      numTestExamples: 10
+      depth: 3
+      numClasses: 4
+      numFeatures: 5
+  - name: classification.GBTClassification
+    params:
+      numExamples: 100
+      numTestExamples: 10
+      depth: 3
+      numClasses: 4
+      numFeatures: 5
+      maxIter: 3