public class RandomForestRegressionModel extends PredictionModel<Vector,RandomForestRegressionModel> implements RandomForestRegressorParams, TreeEnsembleModel<DecisionTreeRegressionModel>, MLWritable, scala.Serializable
param: _trees Decision trees in the ensemble. param: numFeatures Number of features used by this model
| Modifier and Type | Method and Description |
|---|---|
RandomForestRegressionModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
Vector |
featureImportances()
Estimate of the importance of each feature.
|
static RandomForestRegressionModel |
load(String path) |
int |
numFeatures()
Returns the number of features the model was trained on.
|
double |
predict(Vector features)
Predict label for the given features.
|
static MLReader<RandomForestRegressionModel> |
read() |
String |
toString()
Summary of the model
|
DecisionTreeRegressionModel[] |
trees()
Trees in this ensemble.
|
double[] |
treeWeights()
Weights for each tree, zippable with
trees |
String |
uid()
An immutable unique ID for the object and its derivatives.
|
MLWriter |
write()
Returns an
MLWriter instance for this ML instance. |
setFeaturesCol, setPredictionCol, transform, transformSchematransform, transform, transformgetNumTrees, numTrees, setNumTreesfeatureSubsetStrategy, getFeatureSubsetStrategy, getOldStrategy, getSubsamplingRate, setFeatureSubsetStrategy, setSubsamplingRate, subsamplingRatecacheNodeIds, getCacheNodeIds, getMaxBins, getMaxDepth, getMaxMemoryInMB, getMinInfoGain, getMinInstancesPerNode, getOldStrategy, maxBins, maxDepth, maxMemoryInMB, minInfoGain, minInstancesPerNode, setCacheNodeIds, setCheckpointInterval, setMaxBins, setMaxDepth, setMaxMemoryInMB, setMinInfoGain, setMinInstancesPerNode, setSeedvalidateAndTransformSchemagetLabelCol, labelColfeaturesCol, getFeaturesColgetPredictionCol, predictionColclear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwncheckpointInterval, getCheckpointIntervalgetImpurity, getOldImpurity, impurity, setImpurityjavaTreeWeights, toDebugString, totalNumNodessaveinitializeLogging, initializeLogIfNecessary, initializeLogIfNecessary, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarningpublic static MLReader<RandomForestRegressionModel> read()
public static RandomForestRegressionModel load(String path)
public String uid()
Identifiableuid in interface Identifiablepublic int numFeatures()
PredictionModelnumFeatures in class PredictionModel<Vector,RandomForestRegressionModel>public DecisionTreeRegressionModel[] trees()
TreeEnsembleModeltrees in interface TreeEnsembleModel<DecisionTreeRegressionModel>public double[] treeWeights()
TreeEnsembleModeltreestreeWeights in interface TreeEnsembleModel<DecisionTreeRegressionModel>public double predict(Vector features)
PredictionModeltransform() and output predictionCol.predict in class PredictionModel<Vector,RandomForestRegressionModel>features - (undocumented)public RandomForestRegressionModel copy(ParamMap extra)
ParamsdefaultCopy().copy in interface Paramscopy in class Model<RandomForestRegressionModel>extra - (undocumented)public String toString()
TreeEnsembleModeltoString in interface TreeEnsembleModel<DecisionTreeRegressionModel>toString in interface IdentifiabletoString in class Objectpublic Vector featureImportances()
Each feature's importance is the average of its importance across all trees in the ensemble The importance vector is normalized to sum to 1. This method is suggested by Hastie et al. (Hastie, Tibshirani, Friedman. "The Elements of Statistical Learning, 2nd Edition." 2001.) and follows the implementation from scikit-learn.
DecisionTreeRegressionModel.featureImportancespublic MLWriter write()
MLWritableMLWriter instance for this ML instance.write in interface MLWritable