Merge pull request #597 from mmaksi/main

The predict function must take X_test not y_test
mrdbourke · Oct 12, 2023 · bdc7b60 · bdc7b60
2 parents d7cde3d + 8da0981
commit bdc7b60
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/02_neural_network_classification_in_tensorflow.ipynb b/02_neural_network_classification_in_tensorflow.ipynb
@@ -1798,7 +1798,7 @@
       },
       "source": [
         "# Make predictions with our trained model\n",
-        "y_reg_preds = model_3.predict(y_reg_test)\n",
+        "y_reg_preds = model_3.predict(X_reg_test)\n",
         "\n",
         "# Plot the model's predictions against our regression data\n",
         "plt.figure(figsize=(10, 7))\n",
@@ -6874,4 +6874,4 @@
       ]
     }
   ]
-}
+}