LogisticRegressionModel

class snap_ml_spark.estimator.LogisticRegressionModel(java_model=None)

Model fitted by LogisticRegression.

property coefficientMatrix

Model coefficients.

property coefficients

Model coefficients of binomial logistic regression. An exception is thrown in the case of multinomial logistic regression.

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 of binomial logistic regression. An exception is thrown in the case of multinomial logistic regression.

property interceptVector

Model intercept.

property summary

Gets summary (e.g. accuracy/precision/recall, objective history, total iterations) of model trained on the training set. An exception is thrown if trainingSummary is None.