Skip to content

Commit a993ce3

Browse files
committed
increased fraction within tolerance for decision tree regressor test suite
1 parent a8ebf22 commit a993ce3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mllib/src/test/scala/org/apache/spark/ml/regression/DecisionTreeRegressorSuite.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,10 +177,10 @@ class DecisionTreeRegressorSuite extends MLTest with DefaultReadWriteTest {
177177
.setSeed(123)
178178
MLTestingUtils.testArbitrarilyScaledWeights[DecisionTreeRegressionModel,
179179
DecisionTreeRegressor](df.as[LabeledPoint], estimator,
180-
MLTestingUtils.modelPredictionEquals(df, _ ~= _ relTol 0.05, 0.9))
180+
MLTestingUtils.modelPredictionEquals(df, _ ~= _ relTol 0.1, 0.99))
181181
MLTestingUtils.testOutliersWithSmallWeights[DecisionTreeRegressionModel,
182182
DecisionTreeRegressor](df.as[LabeledPoint], estimator, numClasses,
183-
MLTestingUtils.modelPredictionEquals(df, _ ~= _ relTol 0.1, 0.8),
183+
MLTestingUtils.modelPredictionEquals(df, _ ~= _ relTol 0.1, 0.99),
184184
outlierRatio = 2)
185185
MLTestingUtils.testOversamplingVsWeighting[DecisionTreeRegressionModel,
186186
DecisionTreeRegressor](df.as[LabeledPoint], estimator,

0 commit comments

Comments
 (0)