Packages

class LinearRegressionModel extends GeneralizedLinearModel with RegressionModel with Serializable with Saveable with PMMLExportable

Regression model trained using LinearRegression.

Annotations
@Since( "0.8.0" )
Source
LinearRegression.scala
Linear Supertypes
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. LinearRegressionModel
  2. PMMLExportable
  3. Saveable
  4. RegressionModel
  5. GeneralizedLinearModel
  6. Serializable
  7. Serializable
  8. AnyRef
  9. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new LinearRegressionModel(weights: Vector, intercept: Double)

    weights

    Weights computed for every feature.

    intercept

    Intercept computed for this model.

    Annotations
    @Since( "1.1.0" )

Value Members

  1. val intercept: Double
    Definition Classes
    LinearRegressionModelGeneralizedLinearModel
    Annotations
    @Since( "0.8.0" )
  2. def predict(testData: JavaRDD[Vector]): JavaRDD[Double]

    Predict values for examples stored in a JavaRDD.

    Predict values for examples stored in a JavaRDD.

    testData

    JavaRDD representing data points to be predicted

    returns

    a JavaRDD[java.lang.Double] where each entry contains the corresponding prediction

    Definition Classes
    RegressionModel
    Annotations
    @Since( "1.0.0" )
  3. def predict(testData: Vector): Double

    Predict values for a single data point using the model trained.

    Predict values for a single data point using the model trained.

    testData

    array representing a single data point

    returns

    Double prediction from the trained model

    Definition Classes
    GeneralizedLinearModel
    Annotations
    @Since( "1.0.0" )
  4. def predict(testData: RDD[Vector]): RDD[Double]

    Predict values for the given data set using the model trained.

    Predict values for the given data set using the model trained.

    testData

    RDD representing data points to be predicted

    returns

    RDD[Double] where each entry contains the corresponding prediction

    Definition Classes
    GeneralizedLinearModel
    Annotations
    @Since( "1.0.0" )
  5. def save(sc: SparkContext, path: String): Unit

    Save this model to the given path.

    Save this model to the given path.

    This saves:

    • human-readable (JSON) model metadata to path/metadata/
    • Parquet formatted data to path/data/

    The model may be loaded using Loader.load.

    sc

    Spark context used to save model data.

    path

    Path specifying the directory in which to save this model. If the directory already exists, this method throws an exception.

    Definition Classes
    LinearRegressionModelSaveable
    Annotations
    @Since( "1.3.0" )
  6. def toPMML(): String

    Export the model to a String in PMML format

    Export the model to a String in PMML format

    Definition Classes
    PMMLExportable
    Annotations
    @Since( "1.4.0" )
  7. def toPMML(outputStream: OutputStream): Unit

    Export the model to the OutputStream in PMML format

    Export the model to the OutputStream in PMML format

    Definition Classes
    PMMLExportable
    Annotations
    @Since( "1.4.0" )
  8. def toPMML(sc: SparkContext, path: String): Unit

    Export the model to a directory on a distributed file system in PMML format

    Export the model to a directory on a distributed file system in PMML format

    Definition Classes
    PMMLExportable
    Annotations
    @Since( "1.4.0" )
  9. def toPMML(localPath: String): Unit

    Export the model to a local file in PMML format

    Export the model to a local file in PMML format

    Definition Classes
    PMMLExportable
    Annotations
    @Since( "1.4.0" )
  10. def toString(): String

    Print a summary of the model.

    Print a summary of the model.

    Definition Classes
    GeneralizedLinearModel → AnyRef → Any
  11. val weights: Vector
    Definition Classes
    LinearRegressionModelGeneralizedLinearModel
    Annotations
    @Since( "1.0.0" )