categorical[np.arange(n), y] = 1 IndexError: index 2 is out of bounds for axis 1 with size 2