y = np.eye(y.max() + 1)[y].astype('float32')