Package org.apache.spark.ml.tree

Interface GBTRegressorParams

All Superinterfaces:

DecisionTreeParams, GBTParams, HasCheckpointInterval, HasFeaturesCol, HasLabelCol, HasMaxIter, HasPredictionCol, HasSeed, HasStepSize, HasValidationIndicatorCol, HasVarianceImpurity, HasWeightCol, Identifiable, Params, PredictorParams, Serializable, TreeEnsembleParams, TreeEnsembleRegressorParams, TreeRegressorParams

All Known Implementing Classes:

GBTRegressionModel, GBTRegressor

public interface GBTRegressorParams
extends GBTParams, TreeEnsembleRegressorParams, TreeRegressorParams

Method Summary

Modifier and Type	Method	Description
Loss	convertToOldLossType(String loss)
String	getLossType()
Loss	getOldLossType()	(private[ml]) Convert new loss to old loss.
Param<String>	lossType()	Loss function which GBT tries to minimize.

Methods inherited from interface org.apache.spark.ml.tree.DecisionTreeParams

cacheNodeIds, getCacheNodeIds, getLeafCol, getMaxBins, getMaxDepth, getMaxMemoryInMB, getMinInfoGain, getMinInstancesPerNode, getMinWeightFractionPerNode, getOldStrategy, leafCol, maxBins, maxDepth, maxMemoryInMB, minInfoGain, minInstancesPerNode, minWeightFractionPerNode, setLeafCol

Methods inherited from interface org.apache.spark.ml.tree.GBTParams

getOldBoostingStrategy, getValidationTol, stepSize, validationTol

Methods inherited from interface org.apache.spark.ml.param.shared.HasCheckpointInterval

checkpointInterval, getCheckpointInterval

Methods inherited from interface org.apache.spark.ml.param.shared.HasFeaturesCol

featuresCol, getFeaturesCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasLabelCol

getLabelCol, labelCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasMaxIter

getMaxIter, maxIter

Methods inherited from interface org.apache.spark.ml.param.shared.HasPredictionCol

getPredictionCol, predictionCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasSeed

getSeed, seed

Methods inherited from interface org.apache.spark.ml.param.shared.HasStepSize

getStepSize

Methods inherited from interface org.apache.spark.ml.param.shared.HasValidationIndicatorCol

getValidationIndicatorCol, validationIndicatorCol

Methods inherited from interface org.apache.spark.ml.tree.HasVarianceImpurity

getImpurity, getOldImpurity, impurity

Methods inherited from interface org.apache.spark.ml.param.shared.HasWeightCol

getWeightCol, weightCol

Methods inherited from interface org.apache.spark.ml.util.Identifiable

toString, uid

Methods inherited from interface org.apache.spark.ml.param.Params

clear, copy, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, onParamChange, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn

Methods inherited from interface org.apache.spark.ml.tree.TreeEnsembleParams

featureSubsetStrategy, getFeatureSubsetStrategy, getOldStrategy, getSubsamplingRate, subsamplingRate

Methods inherited from interface org.apache.spark.ml.tree.TreeEnsembleRegressorParams

validateAndTransformSchema

Method Details

lossType

Param<String> lossType()

Loss function which GBT tries to minimize. (case-insensitive) Supported: "squared" (L2) and "absolute" (L1) (default = squared)

Returns:

(undocumented)

getLossType

String getLossType()

getOldLossType

Loss getOldLossType()

(private[ml]) Convert new loss to old loss.

Specified by:

getOldLossType in interface GBTParams

convertToOldLossType

Loss convertToOldLossType(String loss)