Method predict

spark/src/main/java/io/ddf/spark/ml/Model.java:16–32

(double[] point)

  }

  @Override
  public Double predict(double[] point) throws DDFException {
    MLClassMethods.PredictMethod predictMethod= new MLClassMethods.PredictMethod(this.getRawModel(), MLClassMethods.DEFAULT_PREDICT_METHOD_NAME,
      new Class<?>[]{Vector.class});
    if(predictMethod.getMethod() == null) {
      throw new DDFException(String.format("Cannot locate method specified by %s", MLClassMethods.DEFAULT_PREDICT_METHOD_NAME));

    }
    Object prediction = predictMethod.instanceInvoke(Vectors.dense(point));
    if(prediction instanceof Double) {
      return (Double) prediction;
    } else if (prediction instanceof Integer) {
      return ((Integer) prediction).doubleValue();
    } else {
      throw new DDFException(String.format("Error getting prediction from model %s", this.getRawModel().getClass().getName()));
    }
  }
}