The predict function must take X_test not y_test

mrdbourke / tensorflow-deep-learning

All course materials for the Zero to Mastery Deep Learning with TensorFlow course.

https://dbourke.link/ZTMTFcourse

MIT License

5.05k stars 2.5k forks source link

The predict function must take X_test not y_test #597

Closed mmaksi closed 9 months ago

mmaksi commented 9 months ago

By mistake, the code says model.predict(y_test) while it should say model.predict(X_test) because the predict function takes the X_test data to output y_preds to be later compared to y_test and measure the accuracy of the model.

mrdbourke commented 9 months ago

Thank you very much for this!

This fixes: https://github.com/mrdbourke/tensorflow-deep-learning/issues/599