Class: DecisionTreeRegressor

eclairjs/ml/regression. DecisionTreeRegressor

Decision tree learning algorithm for regression. It supports both continuous and categorical features.

Constructor

new DecisionTreeRegressor(uidopt)

Parameters:
Name Type Attributes Description
uid string <optional>
Source:

Extends

Methods

(static) load(path) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
path string
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

(static) supportedImpurities() → {Array.<string>}

Accessor for supported impurities: variance
Source:
Returns:
Type
Array.<string>

copy(extra) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
extra module:eclairjs/ml/param.ParamMap
Overrides:
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

extractParamMap() → {module:eclairjs/ml/param.ParamMap}

Inherited From:
Source:
Returns:
Type
module:eclairjs/ml/param.ParamMap

featuresCol() → {module:eclairjs/ml/param.Param}

Param for features column name.
Source:
Returns:
Type
module:eclairjs/ml/param.Param

fit(dataset, paramMapopt) → {module:eclairjs/ml.Estimator|module:eclairjs/ml/feature.Bucketizer}

Fits a model to the input data.
Parameters:
Name Type Attributes Description
dataset module:eclairjs/sql.Dataset
paramMap module:eclairjs/ml/param.ParamMap <optional>
 Parameter map. These values override any specified in this Estimator's embedded ParamMap.
Inherited From:
Source:
Returns:
fitted model
Type
module:eclairjs/ml.Estimator | module:eclairjs/ml/feature.Bucketizer

getFeaturesCol() → {string}

Source:
Returns:
Type
string

getLabelCol() → {string}

Source:
Returns:
Type
string

getPredictionCol() → {string}

Source:
Returns:
Type
string

labelCol() → {module:eclairjs/ml/param.Param}

Param for label column name.
Source:
Returns:
Type
module:eclairjs/ml/param.Param

predictionCol() → {module:eclairjs/ml/param.Param}

Param for prediction column name.
Source:
Returns:
Type
module:eclairjs/ml/param.Param

setCacheNodeIds(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value boolean
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setCheckpointInterval(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value integer
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setFeaturesCol(value) → {module:eclairjs/ml.Predictor}

Parameters:
Name Type Description
value string
Inherited From:
Source:
Returns:
Type
module:eclairjs/ml.Predictor

setImpurity(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value string
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setLabelCol(value) → {module:eclairjs/ml.Predictor}

Parameters:
Name Type Description
value string
Inherited From:
Source:
Returns:
Type
module:eclairjs/ml.Predictor

setMaxBins(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value integer
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setMaxDepth(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value integer
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setMaxMemoryInMB(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value integer
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setMinInfoGain(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value float
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setMinInstancesPerNode(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value integer
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setPredictionCol(value) → {module:eclairjs/ml.Predictor}

Parameters:
Name Type Description
value string
Inherited From:
Source:
Returns:
Type
module:eclairjs/ml.Predictor

setSeed(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value integer
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setVarianceCol(value) → {module:eclairjs/ml/regression.DecisionTreeRegressor}

Parameters:
Name Type Description
value string
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

transformSchema(schema) → {module:eclairjs/sql/types.StructType}

Parameters:
Name Type Description
schema module:eclairjs/sql/types.StructType
Inherited From:
Source:
Returns:
Type
module:eclairjs/sql/types.StructType

uid() → {string}

An immutable unique ID for the object and its derivatives.
Source:
Returns:
Type
string

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
Source:
Returns:
Type
module:eclairjs/sql/types.StructType