在对分类神经网络进行训练时,涉及到标签的设置,分为纯数字类及one-hot类,后者采用的比较多,因此涉及到如何进行标签的转换,如标签([[1],[2],[3]])需要转换成([[1,0,0],[0,1,0],[0,0,1]]),Python程序如下:
import numpy as np
a=np.array([[1],[2],[3]])
a=np.reshape(origin_label,[-1]) #将([[1],[2],[3]])转换为([1,2,3])
b=np.zeros([3,3]) #生成([[0,0,0],[0,0,0],[0,0,0]])
b(a==1,0)=1 #生成([[1,0,0],[0,0,0],[0,0,0]])
b(a==2,1)=1 #生成([[1,0,0],[0,1,0],[0,0,0]])
b(a==3,2)=1 #生成([[1,0,0],[0,1,0],[0,0,1]]),完毕