C2, W4, how to solve this issue

Hi all, how to solve this issue? Thanks.

Please check out np.argmax for how to identify the index of the maximum element.

P.S. I am removing your assignment code as we should not share it.