我的考试和培训目录如下
火车--class_0 --class_1 -测试--class_0 --class_1
到目前为止，所有类都包含2张图像，用于测试和培训。
我正在使用以下代码对输出标签进行一种热编码
def convert_to_one_hot(Y, C):
print ("hehe ",Y.reshape(-1), np.eye(C))
Y = np.eye(C)[Y.reshape(-1)].T
return Y
在主程序中，我传递C = 2和Y = [1 1 2 2]，这样做时出现以下错误：
---> 55 Y = np.eye(C)[Y.reshape(-1)].T
IndexError: index 2 is out of bounds for axis 0 with size 2
我无法纠正此错误。请帮忙！