Class Summary |
DecisionTreeModel |
:: Experimental ::
Decision tree model for classification or regression. |
GradientBoostedTreesModel |
:: Experimental ::
Represents a gradient boosted trees model. |
InformationGainStats |
:: DeveloperApi ::
Information gain statistics for each split
param: gain information gain value
param: impurity current node impurity
param: leftImpurity left node impurity
param: rightImpurity right node impurity
param: leftPredict left node predict
param: rightPredict right node predict |
Node |
:: DeveloperApi ::
Node in a decision tree. |
Predict |
Predicted value for a node
param: predict predicted value
param: prob probability of the label (classification only) |
RandomForestModel |
:: Experimental ::
Represents a random forest model. |
Split |
:: DeveloperApi ::
Split applied to a feature
param: feature feature index
param: threshold Threshold for continuous feature. |