Class: Predictor

eclairjs/ml. Predictor

Abstraction for prediction problems (regression and classification). E.g., VectorUDT for vector features. parameter to specify the concrete type. parameter to specify the concrete type for the corresponding model.

Constructor

new Predictor()

Source:

Extends

Methods

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

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

extractParamMap() → {module:eclairjs/ml/param.ParamMap}

Inherited From:
Source:
Returns:
Type
module:eclairjs/ml/param.ParamMap

fit(dataset, paramMapopt) → {module:eclairjs/ml.Estimator|module:eclairjs/ml/feature.Bucketizer}

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.Estimator | module:eclairjs/ml/feature.Bucketizer

setFeaturesCol(value) → {module:eclairjs/ml.Predictor}

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

setLabelCol(value) → {module:eclairjs/ml.Predictor}

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

setPredictionCol(value) → {module:eclairjs/ml.Predictor}

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

transformSchema(schema) → {module:eclairjs/sql/types.StructType}

Parameters:
Name Type Description
schema module:eclairjs/sql/types.StructType
Overrides:
Source:
Returns:
Type
module:eclairjs/sql/types.StructType