|
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. |