org.apache.lens.ml.algo.spark

Class BaseSparkClassificationModel<MODEL extends org.apache.spark.mllib.classification.ClassificationModel>

java.lang.Object

org.apache.lens.ml.algo.api.MLModel<Double>

org.apache.lens.ml.algo.lib.ClassifierBaseModel

org.apache.lens.ml.algo.spark.BaseSparkClassificationModel<MODEL>

Type Parameters:

MODEL - the generic type

All Implemented Interfaces:

Serializable

Direct Known Subclasses:

DecisionTreeClassificationModel, LogitRegressionClassificationModel, NaiveBayesClassificationModel, SVMClassificationModel

public class BaseSparkClassificationModel<MODEL extends org.apache.spark.mllib.classification.ClassificationModel>
extends ClassifierBaseModel

The Class BaseSparkClassificationModel.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
BaseSparkClassificationModel(String modelId, MODEL model) Instantiates a new base spark classification model.

Method Summary

Methods

Modifier and Type	Method and Description
String	getId() The id.
Double	predict(Object... args) Predict.

Methods inherited from class org.apache.lens.ml.algo.lib.ClassifierBaseModel

getFeatureVector

Methods inherited from class org.apache.lens.ml.algo.api.MLModel

getAlgoName, getCreatedAt, getFeatureColumns, getLabelColumn, getParams, getTable, setAlgoName, setCreatedAt, setFeatureColumns, setId, setLabelColumn, setParams, setTable, toString

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

BaseSparkClassificationModel

public BaseSparkClassificationModel(String modelId,
                            MODEL model)

Instantiates a new base spark classification model.

Parameters:

modelId - the model id

model - the model

Method Detail

predict

public Double predict(Object... args)

Description copied from class: MLModel

Predict.

Specified by:

predict in class MLModel<Double>

Parameters:

args - the args

Returns:

the prediction

getId

public String getId()

Description copied from class: MLModel

The id.

Overrides:

getId in class MLModel<Double>