Unable to plot Decision Boundary in MLP

I got this error while plotting.
ValueError: cannot reshape array of size 377496 into shape (441,428)

plz Help

link to Notebook https://colab.research.google.com/drive/1thUMlWHBYrUlWuDTQ3kEg9nHRoJpR2-S

Hey @last_time, i think you have messed up the shapes, try running the same code on cb.lk/ml18

I hope I’ve cleared your doubt. I ask you to please rate your experience here
Your feedback is very important. It helps us improve our platform and hence provide you
the learning experience you deserve.

On the off chance, you still have some questions or not find the answers satisfactory, you may reopen
the doubt.