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

Class SparkDecisionTreeClassificationOperator.Model

java.lang.Object
org.apache.wayang.spark.operators.ml.SparkDecisionTreeClassificationOperator.Model
All Implemented Interfaces:
Serializable, DecisionTreeClassificationModel, Model, SparkMLModel<double[],Integer>
Enclosing class:
SparkDecisionTreeClassificationOperator
public static class SparkDecisionTreeClassificationOperator.Model extends Object implements DecisionTreeClassificationModel, SparkMLModel<double[],Integer>
See Also:

  • 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<Integer>
    predict(org.apache.spark.api.java.JavaRDD<double[]> input)
     
    org.apache.spark.api.java.JavaRDD<Tuple2<double[],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 Details

    • Model

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

  • Method Details