diff --git a/src/main/scala/configs/mllib-small.yaml b/src/main/scala/configs/mllib-small.yaml index f9896f0..24408f8 100644 --- a/src/main/scala/configs/mllib-small.yaml +++ b/src/main/scala/configs/mllib-small.yaml @@ -65,6 +65,8 @@ benchmarks: maxIter: 3 - name: regression.LinearRegression params: + numExamples: 100 + numTestExamples: 100 numFeatures: 100 regParam: 0.1 tol: [0.0]