org.apache.lens.ml.algo.spark.dt

Class DecisionTreeClassificationModel

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<SparkDecisionTreeModel>

org.apache.lens.ml.algo.spark.dt.DecisionTreeClassificationModel

All Implemented Interfaces:

Serializable

public class DecisionTreeClassificationModel
extends BaseSparkClassificationModel<SparkDecisionTreeModel>

The Class DecisionTreeClassificationModel.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DecisionTreeClassificationModel(String modelId, SparkDecisionTreeModel model) Instantiates a new decision tree classification model.

Method Summary

Methods inherited from class org.apache.lens.ml.algo.spark.BaseSparkClassificationModel

getId, 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

DecisionTreeClassificationModel

public DecisionTreeClassificationModel(String modelId,
                                       SparkDecisionTreeModel model)

Instantiates a new decision tree classification model.

Parameters:

modelId - the model id

model - the model