Getting error in model.fit() at 4:45:46

[Vinayak409]

Hello sir @mrdbourke

In your TensorFlow deep learning tutorial I am getting this error when i am executing model.fit(X, y, epochs=5) , can you please help me to fix this error ?

colab link : https://colab.research.google.com/drive/1c_wm47wRkQZuVlRcdtg2BHTD9KsdUNXp?usp=sharing

[fortesfr]

I think this is answered in discussion #278. The notebook has been updated for TensorFlow 2.7

[fortesfr]

Try adding below after casting to float32
X = tf.expand_dims(X, axis =1)