| Class | Description |
|---|---|
| DecisionTreeModel |
Decision tree model for classification or regression.
|
| GradientBoostedTreesModel |
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 |
Represents a random forest model.
|
| Split |
:: DeveloperApi ::
Split applied to a feature
param: feature feature index
param: threshold Threshold for continuous feature.
|