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

Class SparkDecisionTreeRegressionOperator.Model

java.lang.Object

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

All Implemented Interfaces:

Serializable, DecisionTreeRegressionModel, Model, SparkMLModel<double[],Double>

Enclosing class:

SparkDecisionTreeRegressionOperator

public static class SparkDecisionTreeRegressionOperator.Model
extends Object
implements DecisionTreeRegressionModel, SparkMLModel<double[],Double>

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
Model(org.apache.spark.ml.regression.DecisionTreeRegressionModel model)

Method Summary

Modifier and Type	Method	Description
double	predict(double[] features)	Predict the output value for a single feature vector.
org.apache.spark.api.java.JavaRDD<Double>	predict(org.apache.spark.api.java.JavaRDD<double[]> input)
org.apache.spark.api.java.JavaRDD<Tuple2<double[],Double>>	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 Details

Model

public Model(org.apache.spark.ml.regression.DecisionTreeRegressionModel model)

Method Details

predict

public double predict(double[] features)

Description copied from interface: DecisionTreeRegressionModel

Predict the output value for a single feature vector.

Specified by:

predict in interface DecisionTreeRegressionModel

Parameters:

features - the input features

Returns:

the predicted value

predict

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

Specified by:

predict in interface SparkMLModel<double[],Double>

transform

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

Specified by:

transform in interface SparkMLModel<double[],Double>