Skip to content

[SPARK-9612][ML][FOLLOWUP] fix GBT support weights if subsamplingRate<1 #27070

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
zhengruifeng wants to merge 4 commits into from
Closed

[SPARK-9612][ML][FOLLOWUP] fix GBT support weights if subsamplingRate<1 #27070

Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
dac8f69
init
zhengruifeng Jan 2, 2020
a41bcba
update testsuite
zhengruifeng Jan 2, 2020
e3d9200
update doc
zhengruifeng Jan 3, 2020
53750d4
destory nodeToFeaturesBc
zhengruifeng Jan 3, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
destory nodeToFeaturesBc
  • Loading branch information
@zhengruifeng
zhengruifeng committed Jan 3, 2020
commit 53750d49bd4de04aa9acd75b74912a54afd54a2e
1 change: 1 addition & 0 deletions mllib/src/main/scala/org/apache/spark/ml/tree/impl/RandomForest.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -625,6 +625,7 @@ private[spark] object RandomForest extends Logging with Serializable {
binsToBestSplit(aggStats, splits, featuresForNode, nodes(nodeIndex))
(nodeIndex, (split, stats))
}.collectAsMap()
nodeToFeaturesBc.destroy()

timer.stop("chooseSplits")

Expand Down