Github user imatiach-msft commented on a diff in the pull request:
https://github.com/apache/spark/pull/21632#discussion_r220413857
--- Diff:
mllib/src/test/scala/org/apache/spark/ml/regression/DecisionTreeRegressorSuite.scala
---
@@ -159,6 +169,31 @@ class DecisionTreeRegressorSuite extends MLTest with
DefaultReadWriteTest {
}
}
+ test("training with sample weights") {
+ val df = linearRegressionData
+ val numClasses = 0
+ val testParams = Seq(5, 10)
+ for (maxDepth <- testParams) {
+ val estimator = new DecisionTreeRegressor()
+ .setImpurity("variance")
+ .setMaxDepth(maxDepth)
+ .setMaxBins(100)
+ .setMinWeightFractionPerNode(0.05)
+ .setSeed(123)
+ // TODO: This doesn't look correct, need to debug to figure out what
is wrong with tolerance
--- End diff --
fixed after some debugging
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
