b = np.zeros((584,2)) b[np.arange(584), np.asarray(y_test)] = 1 error: IndexError: shape mismatch: indexing arrays could not be broadcast together with shapes (584,) (130,)