From 813bd8ad59e17d65a33ed4d37612dcd7626325eb Mon Sep 17 00:00:00 2001
From: Timothy Hunter <timhunter@databricks.com>
Date: Fri, 1 Jul 2016 10:34:42 -0700
Subject: [PATCH] adding more experiments

---
 src/main/scala/configs/mllib-small.yaml | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/src/main/scala/configs/mllib-small.yaml b/src/main/scala/configs/mllib-small.yaml
index 5d4fac7..e600554 100644
--- 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