Package org.apache.wayang.spark.operators.ml

Class SparkDecisionTreeClassificationOperator.Model

java.lang.Object

org.apache.wayang.spark.operators.ml.SparkDecisionTreeClassificationOperator.Model

All Implemented Interfaces:

java.io.Serializable, DecisionTreeClassificationModel, Model, SparkMLModel<double[],java.lang.Integer>

Enclosing class:

SparkDecisionTreeClassificationOperator

public static class SparkDecisionTreeClassificationOperator.Model
extends java.lang.Object
implements DecisionTreeClassificationModel, SparkMLModel<double[],java.lang.Integer>

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor	Description
Model(org.apache.spark.ml.classification.DecisionTreeClassificationModel model)

Method Summary

Modifier and Type	Method	Description
int	getDepth()
org.apache.spark.api.java.JavaRDD<java.lang.Integer>	predict(org.apache.spark.api.java.JavaRDD<double[]> input)
org.apache.spark.api.java.JavaRDD<Tuple2<double[],java.lang.Integer>>	transform(org.apache.spark.api.java.JavaRDD<double[]> input)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Model

public Model(org.apache.spark.ml.classification.DecisionTreeClassificationModel model)

Method Detail

transform

public org.apache.spark.api.java.JavaRDD<Tuple2<double[],java.lang.Integer>> transform(org.apache.spark.api.java.JavaRDD<double[]> input)

Specified by:

transform in interface SparkMLModel<double[],java.lang.Integer>

predict

public org.apache.spark.api.java.JavaRDD<java.lang.Integer> predict(org.apache.spark.api.java.JavaRDD<double[]> input)

Specified by:

predict in interface SparkMLModel<double[],java.lang.Integer>

getDepth

public int getDepth()

Specified by:

getDepth in interface DecisionTreeClassificationModel