diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py index d08e241b41d23..8fe5b8d8473aa 100644 --- a/python/pyspark/ml/regression.py +++ b/python/pyspark/ml/regression.py @@ -283,11 +283,11 @@ class LinearRegression( >>> model_path = temp_path + "/lr_model" >>> model.save(model_path) >>> model2 = LinearRegressionModel.load(model_path) - >>> model.coefficients[0] == model2.coefficients[0] + >>> bool(model.coefficients[0] == model2.coefficients[0]) True - >>> model.intercept == model2.intercept + >>> bool(model.intercept == model2.intercept) True - >>> model.transform(test0).take(1) == model2.transform(test0).take(1) + >>> bool(model.transform(test0).take(1) == model2.transform(test0).take(1)) True >>> model.numFeatures 1