AssertionError: Squeezed predicted y shape = (32, 64) whereas actual y shape = (32,)

Does this help?