Skip to content

[SPARK-9478] [ml] Add class weights to Random Forest #13851

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
n-triple-a wants to merge 18 commits into from
Closed

[SPARK-9478] [ml] Add class weights to Random Forest #13851

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
bc7e824
Modify impurity implementations, NOTICE: a further modification is ne…
Jun 7, 2016
df3b4e7
save changes, but compile error exists
Jun 7, 2016
7bcabda
simple testSuites and run properly
Jun 7, 2016
61c4858
add unbalenced data test case, modify label prediction process s.t. t…
Jun 7, 2016
aeb0856
Make Decision Tree predicions correct without changing the base class…
Jun 7, 2016
4dc3e32
1.put classweight def to the right area 2.change interfaces of getOld…
Jun 10, 2016
ad55f6d
add SetClassWeights to RandomForestClassifier
Jun 15, 2016
c17067f
change code style such that requirements are met
Jun 15, 2016
17635d9
random forest with class weights runs properly
Jun 15, 2016
9d52c1f
minor changes
Jun 16, 2016
bf1acfd
add Strategy new param doc
Jun 17, 2016
2baf814
move classW def to DeTrParam class
Jun 17, 2016
fe3819c
remove @BeanProperty
Jun 17, 2016
fd2eee5
change getOldImpu interfaces
Jun 17, 2016
455c47e
first version that pass all run_test, by adding redundant constructor…
Jun 20, 2016
9c99973
first version that pass all tests
Jun 21, 2016
f53a2cc
minor modifications for code styling
Jun 21, 2016
ec5f6df
Minor code style modificaions
Jun 22, 2016
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
Next Next commit
add Strategy new param doc
  • Loading branch information
Yuewei Na authored and Yuewei Na committed Jun 17, 2016
commit bf1acfdb7293ee63b31438972c25de683c852e7d
2 changes: 1 addition & 1 deletion mllib/src/main/scala/org/apache/spark/ml/regression/RandomForestRegressor.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ class RandomForestRegressor @Since("1.4.0") (@Since("1.4.0") override val uid: S
val oldDataset: RDD[LabeledPoint] = extractLabeledPoints(dataset)
val strategy =
super.getOldStrategy(categoricalFeatures, numClasses = 0,
OldAlgo.Regression, getOldImpurity, classWeights = Array(1, 1))
OldAlgo.Regression, getOldImpurity, Array())

val instr = Instrumentation.create(this, oldDataset)
instr.logParams(params: _*)
Expand Down
2 changes: 2 additions & 0 deletions mllib/src/main/scala/org/apache/spark/mllib/tree/configuration/Strategy.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -66,6 +66,8 @@ import org.apache.spark.mllib.tree.impurity.{Entropy, Gini, Impurity, Variance,
* E.g. 10 means that the cache will get checkpointed every 10 updates. If
* the checkpoint directory is not set in
* [[org.apache.spark.SparkContext]], this setting is ignored.
* @param classWeights Weights of classes used in classification problems. It will be ignored in
* regression problems.
*/
@Since("1.0.0")
class Strategy @Since("1.3.0") (
Expand Down