Interface TreeClassifierParams

All Superinterfaces:
Identifiable, Params, Serializable, scala.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 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)
    • getImpurity

      String getImpurity()
    • getOldImpurity

      Impurity getOldImpurity()
      Convert new impurity to old impurity.