mrdbourke / tensorflow-deep-learning

All course materials for the Zero to Mastery Deep Learning with TensorFlow course.
https://dbourke.link/ZTMTFcourse
MIT License
5.14k stars 2.53k forks source link

Error in evaluation metrics in time series notebook for Turkey model #378

Closed JTanner4 closed 2 years ago

JTanner4 commented 2 years ago

When evaluating the turkey model it looks like there is a shape problem causing the MAE and other metric results from evaluate_preds() to appear higher than it should be at the end of the time series notebook. I believe that tf.squeeze() should be used on y_test

turkey_results = evaluate_preds(y_true=tf.squeeze(y_test), y_pred = turkey_preds)