diff --git a/GAMA_AutoML_Tutorial.ipynb b/GAMA_AutoML_Tutorial.ipynb index 5c30a9f092c0218a59c9cb096de18829839c0580..30923768aa27246591ed337fe63de54e85ba72e9 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",