fix deprecated load_boston from scikitlearn

google · Nov 22, 2023 · ce293e4 · ce293e4
1 parent 35c2d1f
commit ce293e4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/constrained/nmf.py b/examples/constrained/nmf.py
@@ -101,7 +101,7 @@ def main(argv):
   del argv
 
   # Prepare data.
-  X, _ = datasets.load_boston(return_X_y=True)
+  X, _ = datasets.load_diabetes(return_X_y=True)
   X = jnp.sqrt(X ** 2)
 
   n_samples = X.shape[0]

diff --git a/examples/implicit_diff/lasso_implicit_diff.py b/examples/implicit_diff/lasso_implicit_diff.py
@@ -79,7 +79,7 @@ def main(argv):
   print("Unrolling:", FLAGS.unrolling)
 
   # Prepare data.
-  X, y = datasets.load_boston(return_X_y=True)
+  X, y = datasets.load_diabetes(return_X_y=True)
   X = preprocessing.normalize(X)
   # data = (X_tr, X_val, y_tr, y_val)
   data = model_selection.train_test_split(X, y, test_size=0.33, random_state=0)

diff --git a/examples/implicit_diff/ridge_reg_implicit_diff.py b/examples/implicit_diff/ridge_reg_implicit_diff.py
@@ -71,7 +71,7 @@ def main(argv):
   del argv
 
   # Prepare data.
-  X, y = datasets.load_boston(return_X_y=True)
+  X, y = datasets.load_diabetes(return_X_y=True)
   X = preprocessing.normalize(X)
   # data = (X_tr, X_val, y_tr, y_val)
   data = model_selection.train_test_split(X, y, test_size=0.33, random_state=0)