GitHub user mengxr opened a pull request:
https://github.com/apache/spark/pull/4493
[SPARK-5597][MLLIB] save/load for decision trees and emsembles
This is based on #4444 from @jkbradley with the following changes:
1. Node schema updated to
~~~
treeId: int
nodeId: Int
predict/
|- predict: Double
|- prob: Double
impurity: Double
isLeaf: Boolean
split/
|- feature: Int
|- threshold: Double
|- featureType: Int
|- categories: Array[Double]
leftNodeId: Integer
rightNodeId: Integer
infoGain: Double
~~~
2. Some refactor of the implementation.
You can merge this pull request into a Git repository by running:
$ git pull https://github.com/mengxr/spark SPARK-5597
Alternatively you can review and apply these changes as the patch at:
https://github.com/apache/spark/pull/4493.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 #4493
----
commit dcdbf85caa6f3dbcf1dfabcc3f26c959ad88b84e
Author: Joseph K. Bradley <[email protected]>
Date: 2015-02-05T20:40:07Z
added save/load for decision tree but need to generalize it to ensembles
commit 1d4c264b869884795e4e805d287bb90d9cd82721
Author: Joseph K. Bradley <[email protected]>
Date: 2015-02-06T22:29:26Z
Added save/load for tree ensembles
commit 45873a2ebb9315b2e2cf0dfcc316ec5dd0c5b25e
Author: Joseph K. Bradley <[email protected]>
Date: 2015-02-06T22:34:03Z
org imports
commit eb6e498b94a2c343a44cd351a3998e740c4e5760
Author: Xiangrui Meng <[email protected]>
Date: 2015-02-10T02:21:29Z
update tree export schema and refactor the implementation
----
---
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.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
