Class: GBTClassifier

eclairjs/ml/classification.GBTClassifier

Gradient-Boosted Trees (GBTs) (http://en.wikipedia.org/wiki/Gradient_boosting) learning algorithm for classification. It supports binary labels, as well as both continuous and categorical features. Note: Multiclass labels are not currently supported. The implementation is based upon: J.H. Friedman. "Stochastic Gradient Boosting." 1999. Notes on Gradient Boosting vs. TreeBoost: - This implementation is for Stochastic Gradient Boosting, not for TreeBoost. - Both algorithms learn tree ensembles by minimizing loss functions. - TreeBoost (Friedman, 1999) additionally modifies the outputs at tree leaf nodes based on the loss function, whereas the original gradient boosting method does not. - We expect to implement TreeBoost in the future: [https://issues.apache.org/jira/browse/SPARK-4240]

Constructor

new GBTClassifier(uidopt)

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

Extends

Methods

(static) load(path) → {GBTClassifier}

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

copy(extra) → {module:eclairjs/ml/classification.GBTClassifier}

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

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/classification.GBTClassifier}

Parameters:
Name Type Description
value boolean
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

setCheckpointInterval(value) → {module:eclairjs/ml/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

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/classification.GBTClassifier}

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/classification.GBTClassifier

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/classification.GBTClassifier}

Parameters:
Name Type Description
value string
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

setMaxBins(value) → {module:eclairjs/ml/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

setMaxDepth(value) → {module:eclairjs/ml/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

setMaxIter(value) → {module:eclairjs/ml/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

setMaxMemoryInMB(value) → {module:eclairjs/ml/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

setMinInfoGain(value) → {module:eclairjs/ml/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

setMinInstancesPerNode(value) → {module:eclairjs/ml/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

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/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

setStepSize(value) → {module:eclairjs/ml/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

setSubsamplingRate(value) → {module:eclairjs/ml/classification.GBTClassifier}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/classification.GBTClassifier

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