From 2f8e5edacc6153f2d19ad86198b2d8f1099b59d3 Mon Sep 17 00:00:00 2001
From: bergma72 <bergma72@msu.edu>
Date: Fri, 5 May 2023 14:13:17 -0400
Subject: [PATCH] Resolve issue #14
---
GAMA_AutoML_Tutorial.ipynb | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/GAMA_AutoML_Tutorial.ipynb b/GAMA_AutoML_Tutorial.ipynb
index 5c30a9f..3092376 100644
--- a/GAMA_AutoML_Tutorial.ipynb
+++ b/GAMA_AutoML_Tutorial.ipynb
@@ -246,12 +246,12 @@
"metadata": {},
"outputs": [],
"source": [
- "from sklearn.datasets import load_boston\n",
+ "from sklearn.datasets import load_diabetes\n",
"from sklearn.model_selection import train_test_split\n",
"from sklearn.metrics import mean_squared_error\n",
"from gama import GamaRegressor\n",
"\n",
- "X, y = load_boston(return_X_y=True)\n",
+ "X, y = load_diabetes(return_X_y=True)\n",
"X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=0)\n",
"\n",
"automl = GamaRegressor(max_total_time=180, store=\"nothing\", n_jobs=-1)\n",
--
GitLab