Value Error at time 4:45:54 in part 1/2 video

[scottdstearns]

Daniel this is so well done. When we got to our first neural net regression I got a value error on the model.fit(X, y, epochs=5) line.

I tried to read the traceback and stack overflow on it, but couldn't make sense of it. Seems like this should work. Attached my colab notebook and the traceback for the error.

Any help or ideas are welcome.

Thank-you!
copy_of_01_neural_network_regression_in_tensorflow.txt

[scottdstearns]

Thanks to @AmanAdastra for answer to Issue opened by @jcr0269 ... Expanding dimensions of X and y fixed the issue:
X = tf.cast(tf.constant(X), dtype=tf.float32)
y = tf.cast(tf.constant(y), dtype=tf.float32)
X = tf.expand_dims(X,axis=1)
y = tf.expand_dims(y,axis=1)

[mrdbourke]

Thank you for the kind words and glad you got it fixed!