|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |
Object org.apache.spark.ml.PipelineStage org.apache.spark.ml.Estimator<M> org.apache.spark.ml.Predictor<Vector,GBTClassifier,GBTClassificationModel> org.apache.spark.ml.classification.GBTClassifier
public final class GBTClassifier
:: Experimental ::
Gradient-Boosted Trees (GBTs)
learning algorithm for classification.
It supports binary labels, as well as both continuous and categorical features.
Note: Multiclass labels are not currently supported.
Constructor Summary | |
---|---|
GBTClassifier()
|
|
GBTClassifier(String uid)
|
Method Summary | |
---|---|
GBTClassifier |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params. |
String |
getLossType()
|
Param<String> |
lossType()
Loss function which GBT tries to minimize. |
GBTClassifier |
setCacheNodeIds(boolean value)
|
GBTClassifier |
setCheckpointInterval(int value)
|
GBTClassifier |
setImpurity(String value)
The impurity setting is ignored for GBT models. |
GBTClassifier |
setLossType(String value)
|
GBTClassifier |
setMaxBins(int value)
|
GBTClassifier |
setMaxDepth(int value)
|
GBTClassifier |
setMaxIter(int value)
|
GBTClassifier |
setMaxMemoryInMB(int value)
|
GBTClassifier |
setMinInfoGain(double value)
|
GBTClassifier |
setMinInstancesPerNode(int value)
|
GBTClassifier |
setSeed(long value)
|
GBTClassifier |
setStepSize(double value)
|
GBTClassifier |
setSubsamplingRate(double value)
|
static String[] |
supportedLossTypes()
Accessor for supported loss settings: logistic |
String |
uid()
|
StructType |
validateAndTransformSchema(StructType schema,
boolean fitting,
DataType featuresDataType)
Validates and transforms the input schema with the provided param map. |
Methods inherited from class org.apache.spark.ml.Predictor |
---|
fit, setFeaturesCol, setLabelCol, setPredictionCol, transformSchema |
Methods inherited from class org.apache.spark.ml.Estimator |
---|
fit, fit, fit, fit |
Methods inherited from class Object |
---|
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait |
Methods inherited from interface org.apache.spark.Logging |
---|
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning |
Methods inherited from interface org.apache.spark.ml.param.Params |
---|
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, setDefault, shouldOwn, validateParams |
Constructor Detail |
---|
public GBTClassifier(String uid)
public GBTClassifier()
Method Detail |
---|
public static final String[] supportedLossTypes()
public String uid()
public GBTClassifier setMaxDepth(int value)
public GBTClassifier setMaxBins(int value)
public GBTClassifier setMinInstancesPerNode(int value)
public GBTClassifier setMinInfoGain(double value)
public GBTClassifier setMaxMemoryInMB(int value)
public GBTClassifier setCacheNodeIds(boolean value)
public GBTClassifier setCheckpointInterval(int value)
public GBTClassifier setImpurity(String value)
value
- (undocumented)
public GBTClassifier setSubsamplingRate(double value)
public GBTClassifier setSeed(long value)
public GBTClassifier setMaxIter(int value)
public GBTClassifier setStepSize(double value)
public Param<String> lossType()
public GBTClassifier setLossType(String value)
public String getLossType()
public GBTClassifier copy(ParamMap extra)
Params
copy
in interface Params
copy
in class Predictor<Vector,GBTClassifier,GBTClassificationModel>
extra
- (undocumented)
defaultCopy()
public StructType validateAndTransformSchema(StructType schema, boolean fitting, DataType featuresDataType)
schema
- input schemafitting
- whether this is in fittingfeaturesDataType
- SQL DataType for FeaturesType.
E.g., VectorUDT
for vector features.
|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |