diff --git a/python/cuml/cuml/tests/test_linear_model.py b/python/cuml/cuml/tests/test_linear_model.py index 042d6b48b7..b676e89360 100644 --- a/python/cuml/cuml/tests/test_linear_model.py +++ b/python/cuml/cuml/tests/test_linear_model.py @@ -1086,7 +1086,7 @@ def test_elasticnet_solvers_eq(datatype, alpha, l1_ratio, nrows, column_info): qn = cuElasticNet(solver="qn", **kwargs) qn.fit(X_train, y_train) # the results of the two models should be close (even if both are bad) - assert qn.score(X_test, cd_res) > 0.95 + assert qn.score(X_test, cd_res) > 0.90 # coefficients of the two models should be close assert np.corrcoef(cd.coef_, qn.coef_)[0, 1] > 0.98