diff --git a/src/main/resources/com/databricks/spark/sql/perf/mllib/config/mllib-large.yaml b/src/main/resources/com/databricks/spark/sql/perf/mllib/config/mllib-large.yaml index 741359d..93e39f4 100644 --- a/src/main/resources/com/databricks/spark/sql/perf/mllib/config/mllib-large.yaml +++ b/src/main/resources/com/databricks/spark/sql/perf/mllib/config/mllib-large.yaml @@ -67,4 +67,26 @@ benchmarks: numItems: 6000000 regParam: 0.01 rank: 10 - maxIter: 10 \ No newline at end of file + maxIter: 10 + - name: regression.DecisionTreeRegression + params: + depth: [5, 10] + - name: regression.GLMRegression + params: + numExamples: 500000 + numTestExamples: 500000 + numFeatures: 1000 + link: log + family: gaussian + tol: 0.0 + maxIter: 10 + regParam: 0.1 + - name: regression.LinearRegression + params: + regParam: 0.01 + tol: 0.0 + maxIter: 20 + - name: regression.RandomForestRegression + params: + depth: 10 + maxIter: 4 \ No newline at end of file