From 7c5bbf6c50b91c0703bedbd0d6e842dbc1c2fd64 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= Date: Sun, 17 Oct 2021 21:33:53 +1300 Subject: [PATCH] Fix off-by-one error, fixes #487 --- 04_training_linear_models.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/04_training_linear_models.ipynb b/04_training_linear_models.ipynb index 223fb48..5eda80a 100644 --- a/04_training_linear_models.ipynb +++ b/04_training_linear_models.ipynb @@ -640,7 +640,7 @@ "def plot_learning_curves(model, X, y):\n", " X_train, X_val, y_train, y_val = train_test_split(X, y, test_size=0.2, random_state=10)\n", " train_errors, val_errors = [], []\n", - " for m in range(1, len(X_train)):\n", + " for m in range(1, len(X_train) + 1):\n", " model.fit(X_train[:m], y_train[:m])\n", " y_train_predict = model.predict(X_train[:m])\n", " y_val_predict = model.predict(X_val)\n",