algorithm for the ensemble model, either Classification or Regression
Current version of model save/load format.
Get number of trees in forest.
Java-friendly version of org.apache.spark.mllib.tree.model.TreeEnsembleModel#predict.
Predict values for the given data set.
RDD representing data points to be predicted
RDD[Double] where each entry contains the corresponding prediction
Predict values for a single data point using the model trained.
array representing a single data point
predicted category from the trained model
Save this model to the given path.
Print the full model to a string.
Print a summary of the model.
Get total number of nodes, summed over all trees in the forest.