Class: GBTRegressor

eclairjs/ml/regression. GBTRegressor

Gradient-Boosted Trees (GBTs) learning algorithm for regression. It supports both continuous and categorical features.

Constructor

new GBTRegressor(uidopt)

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

Extends

Methods

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

Accessor for supported loss settings: squared (L2), absolute (L1)
Source:
Returns:
Type
Array.<string>

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

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

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

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

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

getLossType() → {string}

Source:
Returns:
Type
string

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

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

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

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

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.GBTRegressor}

The impurity setting is ignored for GBT models. Individual trees are built using impurity "Variance."
Parameters:
Name Type Description
value string
Source:
Returns:
Type
module:eclairjs/ml/regression.GBTRegressor

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

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

setLossType(value) → {module:eclairjs/ml/regression.GBTRegressor}

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

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

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

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

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

setMaxIter(value) → {module:eclairjs/ml/regression.GBTRegressor}

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

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

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

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

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

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

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

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.GBTRegressor}

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

setStepSize(value) → {module:eclairjs/ml/regression.GBTRegressor}

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

setSubsamplingRate(value) → {module:eclairjs/ml/regression.GBTRegressor}

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

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