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