LinearRegressionModel

class snap_ml_spark.estimator.LinearRegressionModel(java_model=None)

Model fitted by LinearRegression.

property coefficients

Model coefficients.

evaluate(dataset)

Evaluates the model on a test dataset.

Parameters

dataset – Test dataset to evaluate model on, where dataset is an instance of pyspark.sql.DataFrame

property hasSummary

Indicates whether a training summary exists for this model instance.

property intercept

Model intercept.

property scale

The value will be 1.0.

property summary

Gets summary (e.g. residuals, mse, r-squared ) of model on training set. An exception is thrown if trainingSummary is None.