Interface RegressionModel

All Superinterfaces:
Serializable
All Known Implementing Classes:
LassoModel, LinearRegressionModel, RidgeRegressionModel

public interface RegressionModel extends Serializable
  • Method Summary

    Modifier and Type
    Method
    Description
    predict(JavaRDD<Vector> testData)
    Predict values for examples stored in a JavaRDD.
    double
    predict(Vector testData)
    Predict values for a single data point using the model trained.
    predict(RDD<Vector> testData)
    Predict values for the given data set using the model trained.
  • Method Details

    • predict

      RDD<Object> predict(RDD<Vector> testData)
      Predict values for the given data set using the model trained.

      Parameters:
      testData - RDD representing data points to be predicted
      Returns:
      RDD[Double] where each entry contains the corresponding prediction

    • predict

      double predict(Vector testData)
      Predict values for a single data point using the model trained.

      Parameters:
      testData - array representing a single data point
      Returns:
      Double prediction from the trained model

    • predict

      JavaRDD<Double> predict(JavaRDD<Vector> testData)
      Predict values for examples stored in a JavaRDD.
      Parameters:
      testData - JavaRDD representing data points to be predicted
      Returns:
      a JavaRDD[java.lang.Double] where each entry contains the corresponding prediction