Class: DecisionTreeClassificationModel

eclairjs/ml/classification. DecisionTreeClassificationModel

Decision tree model for classification. It supports both binary and multiclass labels, as well as both continuous and categorical features.

Constructor

new DecisionTreeClassificationModel()

Source:

Methods

(static) load(path) → {module:eclairjs/ml/classification.DecisionTreeClassificationModel}

Parameters:
Name Type Description
path string
Source:
Returns:
Type
module:eclairjs/ml/classification.DecisionTreeClassificationModel

(static) read() → {module:eclairjs/ml/util.MLReader}

Source:
Returns:
Type
module:eclairjs/ml/util.MLReader

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

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

featuresCol() → {module:eclairjs/ml/param.Param}

Param for features column name.
Source:
Returns:
Type
module:eclairjs/ml/param.Param

getFeaturesCol() → {string}

Source:
Returns:
Type
string

getLabelCol() → {string}

Source:
Returns:
Type
string

getPredictionCol() → {string}

Source:
Returns:
Type
string

getRawPredictionCol() → {string}

Source:
Returns:
Type
string

labelCol() → {module:eclairjs/ml/param.Param}

Param for label column name.
Source:
Returns:
Type
module:eclairjs/ml/param.Param

numClasses() → {integer}

Number of classes (values which the label can take).
Source:
Returns:
Type
integer

numFeatures() → {integer}

Returns the number of features the model was trained on. If unknown, returns -1
Source:
Returns:
Type
integer

predictionCol() → {module:eclairjs/ml/param.Param}

Param for prediction column name.
Source:
Returns:
Type
module:eclairjs/ml/param.Param

rawPredictionCol() → {module:eclairjs/ml/param.Param}

Param for raw prediction (a.k.a. confidence) column name.
Source:
Returns:
Type
module:eclairjs/ml/param.Param

rootNode() → {module:eclairjs/ml/tree.Node}

Source:
Returns:
Type
module:eclairjs/ml/tree.Node

toDebugString() → {string}

Source:
Returns:
Type
string

toString() → {string}

Source:
Returns:
Type
string

uid() → {string}

An immutable unique ID for the object and its derivatives.
Source:
Returns:
Type
string

validateAndTransformSchema(schema, fitting, featuresDataType) → {module:eclairjs/sql/types.StructType}

Validates and transforms the input schema with the provided param map.
Parameters:
Name Type Description
schema module:eclairjs/sql/types.StructType
fitting boolean whether this is in fitting
featuresDataType module:eclairjs/sql/types.DataType SQL DataType for FeaturesType. E.g., module:eclairjs/sql/types.VectorUDTfor vector features
Source:
Returns:
Type
module:eclairjs/sql/types.StructType

write() → {module:eclairjs/ml/util.MLWriter}

Source:
Returns:
Type
module:eclairjs/ml/util.MLWriter