Package org.apache.spark.ml.regression

Class LinearRegressionSummary

Object
org.apache.spark.ml.regression.LinearRegressionSummary
All Implemented Interfaces:
Serializable, Summary
Direct Known Subclasses:
LinearRegressionTrainingSummary
public class LinearRegressionSummary extends Object implements Summary, Serializable
Linear regression results evaluated on a dataset.

param: predictions predictions output by the model's transform method. param: predictionCol Field in "predictions" which gives the predicted value of the label at each instance. param: labelCol Field in "predictions" which gives the true label of each instance. param: featuresCol Field in "predictions" which gives the features of each instance as a vector. param: coefficientArray Coefficients of the linear regression model, only necessary when diagInvAtWA is not Array(0).

See Also:

  • Method Details

    • predictions

      public Dataset<Row> predictions()

    • predictionCol

      public String predictionCol()

    • labelCol

      public String labelCol()

    • featuresCol

      public String featuresCol()

    • explainedVariance

      public double explainedVariance()
      Returns the explained variance regression score. explainedVariance = 1 - variance(y - \hat{y}) / variance(y) Reference: Wikipedia explain variation
      Returns:
      (undocumented)

    • meanAbsoluteError

      public double meanAbsoluteError()
      Returns the mean absolute error, which is a risk function corresponding to the expected value of the absolute error loss or l1-norm loss.
      Returns:
      (undocumented)

    • meanSquaredError

      public double meanSquaredError()
      Returns the mean squared error, which is a risk function corresponding to the expected value of the squared error loss or quadratic loss.
      Returns:
      (undocumented)

    • rootMeanSquaredError

      public double rootMeanSquaredError()
      Returns the root mean squared error, which is defined as the square root of the mean squared error.
      Returns:
      (undocumented)

    • r2

      public double r2()
      Returns R^2^, the coefficient of determination. Reference: Wikipedia coefficient of determination
      Returns:
      (undocumented)

    • r2adj

      public double r2adj()
      Returns Adjusted R^2^, the adjusted coefficient of determination. Reference: Wikipedia coefficient of determination
      Returns:
      (undocumented)

    • residuals

      public Dataset<Row> residuals()

    • numInstances

      public long numInstances()

    • degreesOfFreedom

      public long degreesOfFreedom()
      Degrees of freedom

    • devianceResiduals

      public double[] devianceResiduals()

    • coefficientStandardErrors

      public double[] coefficientStandardErrors()

    • tValues

      public double[] tValues()

    • pValues

      public double[] pValues()