adding more experiments

2016-07-01 10:34:42 -07:00 · 2016-07-01 10:34:42 -07:00 · 813bd8ad59
commit 813bd8ad59
parent c15d083fe7
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