[REVIEW] Adding power_t param to SGD failing pytests (#3012)

* adding param to sgd test * changelog and removing stray prints
rapidsai · Oct 18, 2020 · 5beaaae · 5beaaae
1 parent eb1b950
commit 5beaaae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -10,6 +10,7 @@
 
 ## Bug Fixes
 - PR #2983: Fix seeding of KISS99 RNG
+- PR #3012: Increasing learning rate for SGD log loss and invscaling pytests
 
 # cuML 0.16.0 (Date TBD)
 

diff --git a/python/cuml/test/test_sgd.py b/python/cuml/test/test_sgd.py
@@ -51,7 +51,8 @@ def test_sgd(dtype, lrate, penalty, loss, datatype):
 
     cu_sgd = cumlSGD(learning_rate=lrate, eta0=0.005, epochs=2000,
                      fit_intercept=True, batch_size=4096,
-                     tol=0.0, penalty=penalty, loss=loss)
+                     tol=0.0, penalty=penalty, loss=loss,
+                     power_t=0.4)
 
     cu_sgd.fit(X_train, y_train)
     cu_pred = cu_sgd.predict(X_test)