Package org.apache.spark.ml.tree

Interface TreeClassifierParams

All Superinterfaces:

Identifiable, Params, Serializable

All Known Subinterfaces:

DecisionTreeClassifierParams, RandomForestClassifierParams

All Known Implementing Classes:

DecisionTreeClassificationModel, DecisionTreeClassifier, RandomForestClassificationModel, RandomForestClassifier

public interface TreeClassifierParams
extends Params

Parameters for Decision Tree-based classification algorithms.

Method Summary

Modifier and Type	Method	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.util.Identifiable

toString, uid

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

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

Method Details

impurity

Param<String> impurity()

Criterion used for information gain calculation (case-insensitive). This impurity type is used in DecisionTreeClassifier and RandomForestClassifier, Supported: "entropy" and "gini". (default = gini)

Returns:

(undocumented)

getImpurity

String getImpurity()

getOldImpurity

Impurity getOldImpurity()

Convert new impurity to old impurity.