Constructor
new RandomForestRegressionModel()
Extends
Methods
(static) load(path) → {RandomForestRegressionModel}
Parameters:
Name | Type | Description |
---|---|---|
path |
string |
Returns:
- Type
- RandomForestRegressionModel
(static) read() → {MLReader}
Returns:
- Type
- MLReader
copy(extra) → {module:eclairjs/ml/regression.RandomForestRegressionModel}
Parameters:
Name | Type | Description |
---|---|---|
extra |
module:eclairjs/ml/param.ParamMap |
- Overrides:
- Source:
Returns:
featureImportances() → {module:eclairjs/mllib/linalg.Vector}
Estimate of the importance of each feature.
This generalizes the idea of "Gini" importance to other losses, following the explanation of Gini importance
from "Random Forests" documentation by Leo Breiman and Adele Cutler, and following the implementation from scikit-learn.
This feature importance is calculated as follows: - Average over trees: - importance(feature j) = sum (over nodes which split on feature j)
of the gain, where gain is scaled by the number of instances passing through node - Normalize importances for
tree based on total number of training instances used to build tree. - Normalize feature importance vector to sum to 1.
Returns:
featuresCol() → {module:eclairjs/ml/param.Param}
Param for features column name.
- Overrides:
- Source:
Returns:
getFeaturesCol() → {Promise.<string>}
- Overrides:
- Source:
Returns:
- Type
- Promise.<string>
getLabelCol() → {Promise.<string>}
- Overrides:
- Source:
Returns:
- Type
- Promise.<string>
getPredictionCol() → {Promise.<string>}
- Overrides:
- Source:
Returns:
- Type
- Promise.<string>
hasParent() → {Promise.<boolean>}
Indicates whether this Model has a corresponding parent.
- Inherited From:
- Source:
Returns:
- Type
- Promise.<boolean>
labelCol() → {module:eclairjs/ml/param.Param}
Param for label column name.
- Overrides:
- Source:
Returns:
numFeatures() → {Promise.<number>}
Returns the number of features the model was trained on. If unknown, returns -1
- Inherited From:
- Source:
Returns:
- Type
- Promise.<number>
parent() → {module:eclairjs/ml.Estimator}
- Inherited From:
- Source:
Returns:
predictionCol() → {module:eclairjs/ml/param.Param}
Param for prediction column name.
- Overrides:
- Source:
Returns:
setFeaturesCol(value) → {object}
Parameters:
Name | Type | Description |
---|---|---|
value |
string |
- Inherited From:
- Source:
Returns:
- Type
- object
setParent(parent) → {object}
Sets the parent of this model.
Parameters:
Name | Type | Description |
---|---|---|
parent |
module:eclairjs/ml.Estimator |
- Inherited From:
- Source:
Returns:
- Type
- object
setPredictionCol(value) → {object}
Parameters:
Name | Type | Description |
---|---|---|
value |
string |
- Inherited From:
- Source:
Returns:
- Type
- object
toString() → {Promise.<string>}
Returns:
- Type
- Promise.<string>
transform(dataset) → {module:eclairjs/sql.Dataset}
Transforms dataset by reading from featuresCol, calling [[predict()]], and storing
the predictions as a new column predictionCol.
Parameters:
Name | Type | Description |
---|---|---|
dataset |
module:eclairjs/sql.Dataset | input dataset |
- Inherited From:
- Source:
Returns:
transformed dataset with [[predictionCol]] of type [[Double]]
transformSchema(schema) → {module:eclairjs/sql/types.StructType}
Parameters:
Name | Type | Description |
---|---|---|
schema |
module:eclairjs/sql/types.StructType |
- Inherited From:
- Source:
Returns:
trees() → {Array.<module:eclairjs/ml/tree.DecisionTreeRegressionModel>}
Returns:
- Type
- Array.<module:eclairjs/ml/tree.DecisionTreeRegressionModel>
treeWeights() → {Promise.<Array.<number>>}
Returns:
- Type
- Promise.<Array.<number>>
validateAndTransformSchema(schema, fitting, featuresDataType) → {module:eclairjs/sql/types.StructType}
Validates and transforms the input schema with the provided param map.
Parameters:
Name | Type | Description |
---|---|---|
schema |
module:eclairjs/sql/types.StructType | |
fitting |
boolean | whether this is in fitting |
featuresDataType |
module:eclairjs/sql/types.DataType | SQL DataType for FeaturesType. E.g., module:eclairjs/sql/types.VectorUDTfor vector features |
- Overrides:
- Source:
Returns:
write() → {MLWriter}
Returns:
- Type
- MLWriter