Class: ProbabilisticClassifier

eclairjs/ml/classification. ProbabilisticClassifier

Single-label binary or multiclass classifier which can output class conditional probabilities.

Constructor

new ProbabilisticClassifier()

Source:

Extends

Methods

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

Parameters:
Name Type Description
extra module:eclairjs/ml/param.ParamMap
Inherited From:
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

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

Source:
Returns:
Type
module:eclairjs/ml/param.Param

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

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

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

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

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

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

setProbabilityCol(value) → {object}

Parameters:
Name Type Description
value string
Source:
Returns:
Type
object

setRawPredictionCol(value) → {module:eclairjs/ml/classification.Classifier}

Parameters:
Name Type Description
value string
Inherited From:
Source:
Returns:
Type
module:eclairjs/ml/classification.Classifier

setThresholds(value) → {object}

Parameters:
Name Type Description
value Array.<float>
Source:
Returns:
Type
object

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

Source:
Returns:
Type
module:eclairjs/ml/param.Param

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

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