test mypy

online-ml · Dec 1, 2024 · 1daac00 · 1daac00
1 parent 1f2e5f2
commit 1daac00
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/river/linear_model/AUC_SGD.py b/river/linear_model/AUC_SGD.py
@@ -48,10 +48,10 @@ class AUC_SGD:
     >>> theta = np.random.randn(X_train[0].shape[0])
     >>> test_auc = model.getTest(X_train, X_test, y_train)
     >>> train_auc = model.getTrain(X_train, y_train)
-    >>> print(f"{round(roc_auc_score(y_train, train_auc) * 100)}" + "%")
+    >>> print(f"{round(roc_auc_score(y_train, train_auc) * 100)}%")
+    89%
+    >>> print(f"{round(roc_auc_score(y_test, test_auc) * 100)}%")
     89%
-    >>> print(f"{round(roc_auc_score(y_test, test_auc) * 100)}" + "%")
-    88%
     """
 
     def __init__(self, epochs=900, lr=0.5, n_mc=500, gamma=1e-4, eps=0.01):