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

Closed

n-triple-a wants to merge 18 commits intoapache:masterfrom n-triple-a:weightedRandomForest

+428-38

Commits on Jun 7, 2016

Modify impurity implementations, NOTICE: a further modification is needed(getCalculator & fromString method)

Yuewei Na
authored and
Yuewei Na
committed
save changes, but compile error exists

Yuewei Na
authored and
Yuewei Na
committed
simple testSuites and run properly

Yuewei Na
authored and
Yuewei Na
committed
add unbalenced data test case, modify label prediction process s.t. the predictions are correct with Impurity=WeightedGini

Yuewei Na
authored and
Yuewei Na
committed
Make Decision Tree predicions correct without changing the base class ProbClassifier, but changing the definition of Impurity and DecTreClassifier

Yuewei Na
authored and
Yuewei Na
committed

Commits on Jun 10, 2016

1.put classweight def to the right area 2.change interfaces of getOldStrategy 3.make classweights can be passed when reconstructing the tree, including json read/write

Yuewei Na
authored and
Yuewei Na
committed

Commits on Jun 15, 2016

add SetClassWeights to RandomForestClassifier

Yuewei Na
authored and
Yuewei Na
committed
change code style such that requirements are met

Yuewei Na
authored and
Yuewei Na
committed
random forest with class weights runs properly

Yuewei Na
authored and
Yuewei Na
committed

Commits on Jun 16, 2016

minor changes

Yuewei Na
authored and
Yuewei Na
committed

Commits on Jun 17, 2016

add Strategy new param doc

Yuewei Na
authored and
Yuewei Na
committed
move classW def to DeTrParam class

Yuewei Na
authored and
Yuewei Na
committed
remove @BeanProperty

Yuewei Na
authored and
Yuewei Na
committed
change getOldImpu interfaces

Yuewei Na
authored and
Yuewei Na
committed

Commits on Jun 20, 2016

first version that pass all run_test, by adding redundant constructor and reverting getOldStrategy to old versions

Yuewei Na
authored and
Yuewei Na
committed

Commits on Jun 21, 2016

first version that pass all tests

Yuewei Na
authored and
Yuewei Na
committed
minor modifications for code styling

Yuewei Na
authored and
Yuewei Na
committed

Commits on Jun 22, 2016

Minor code style modificaions

Yuewei Na
authored and
Yuewei Na
committed