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) → {DecisionTreeRegressor}

Parameters:
Name Type Description
path string
Source:
Returns:
Type
DecisionTreeRegressor

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

fit(dataset, paramMapopt) → {module:eclairjs/ml.Model}

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.Model

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 number
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 number
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

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

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

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

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

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

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

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

Parameters:
Name Type Description
value number
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 number
Source:
Returns:
Type
module:eclairjs/ml/regression.DecisionTreeRegressor

setVarianceCol(value) → {type}

Parameters:
Name Type Description
value string
Source:
Returns:
Type
type

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() → {Promise.<string>}

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