Github user mtunique commented on a diff in the pull request:
https://github.com/apache/flink/pull/3756#discussion_r142308699
--- Diff:
flink-libraries/flink-ml/src/test/scala/org/apache/flink/ml/regression/MultipleLinearRegressionITSuite.scala
---
@@ -75,15 +75,16 @@ class MultipleLinearRegressionITSuite
val parameters = ParameterMap()
- parameters.add(MultipleLinearRegression.Stepsize, 2.0)
- parameters.add(MultipleLinearRegression.Iterations, 10)
- parameters.add(MultipleLinearRegression.ConvergenceThreshold, 0.001)
+ parameters.add(WithIterativeSolver.Stepsize, 2.0)
+ parameters.add(WithIterativeSolver.Iterations, 10)
+ parameters.add(WithIterativeSolver.ConvergenceThreshold, 0.001)
mlr.fit(sparseInputDS, parameters)
val weightList = mlr.weightsOption.get.collect()
val WeightVector(weights, intercept) = weightList.head
+ println(weightList)
--- End diff --
Ok I will delete it.
---