public final class DecisionTreeClassificationModel extends PredictionModel<Vector,DecisionTreeClassificationModel> implements scala.Serializable
Decision tree
model for classification.
It supports both binary and multiclass labels, as well as both continuous and categorical
features.Modifier and Type | Method and Description |
---|---|
DecisionTreeClassificationModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
static DecisionTreeClassificationModel |
fromOld(DecisionTreeModel oldModel,
DecisionTreeClassifier parent,
scala.collection.immutable.Map<Object,Object> categoricalFeatures)
(private[ml]) Convert a model from the old API
|
Node |
rootNode() |
String |
toString() |
String |
uid() |
StructType |
validateAndTransformSchema(StructType schema,
boolean fitting,
DataType featuresDataType)
Validates and transforms the input schema with the provided param map.
|
setFeaturesCol, setPredictionCol, transform, transformSchema
transform, transform, transform
clear, copyValues, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, setDefault, shouldOwn, validateParams
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public static DecisionTreeClassificationModel fromOld(DecisionTreeModel oldModel, DecisionTreeClassifier parent, scala.collection.immutable.Map<Object,Object> categoricalFeatures)
public String uid()
public Node rootNode()
public DecisionTreeClassificationModel copy(ParamMap extra)
Params
copy
in interface Params
copy
in class Model<DecisionTreeClassificationModel>
extra
- (undocumented)public String toString()
toString
in class Object
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.