org.apache.spark.ml.tree

Interface HasVarianceImpurity

All Superinterfaces:

Identifiable, Params, java.io.Serializable

All Known Subinterfaces:

DecisionTreeRegressorParams, GBTClassifierParams, GBTRegressorParams, RandomForestRegressorParams, TreeRegressorParams

All Known Implementing Classes:

DecisionTreeRegressionModel, DecisionTreeRegressor, GBTClassificationModel, GBTClassifier, GBTRegressionModel, GBTRegressor, RandomForestRegressionModel, RandomForestRegressor

public interface HasVarianceImpurity
extends Params

Method Summary

Modifier and Type	Method and Description
String	getImpurity()
Impurity	getOldImpurity() Convert new impurity to old impurity.
Param<String>	impurity() Criterion used for information gain calculation (case-insensitive).

Methods inherited from interface org.apache.spark.ml.param.Params

clear, copy, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, onParamChange, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn

Methods inherited from interface org.apache.spark.ml.util.Identifiable

toString, uid

Method Detail

impurity

Param<String> impurity()

Criterion used for information gain calculation (case-insensitive). This impurity type is used in DecisionTreeRegressor, RandomForestRegressor, GBTRegressor and GBTClassifier (since GBTClassificationModel is internally composed of DecisionTreeRegressionModels). Supported: "variance". (default = variance)

Returns:

(undocumented)

getImpurity

String getImpurity()

getOldImpurity

Impurity getOldImpurity()

Convert new impurity to old impurity.