GitHub user johnnywalleye opened a pull request:
https://github.com/apache/spark/pull/1343
[SPARK-2417][MLlib] Fix DecisionTree tests
Fixes test failures introduced by https://github.com/apache/spark/pull/1316.
For both the regression and classification cases,
val stats is the InformationGainStats for the best tree split.
stats.predict is the predicted value for the data, before the split is made.
Since 600 of the 1,000 values generated by
DecisionTreeSuite.generateCategoricalDataPoints() are 1.0 and the rest 0.0, the
regression tree and classification tree both correctly predict a value of 0.6
for this data now, and the assertions have been changed to reflect that.
You can merge this pull request into a Git repository by running:
$ git pull https://github.com/johnnywalleye/spark decision-tree-tests
Alternatively you can review and apply these changes as the patch at:
https://github.com/apache/spark/pull/1343.patch
To close this pull request, make a commit to your master/trunk branch
with (at least) the following in the commit message:
This closes #1343
----
commit ef806035dac2971920b697d7a893e27288d87ef1
Author: johnnywalleye <[email protected]>
Date: 2014-07-09T13:25:37Z
[SPARK-2417][MLlib] Fix DecisionTree tests
----
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at [email protected] or file a JIRA ticket
with INFRA.
---
