1. 程式人生 > 其它 >python資料處理——數值轉one-hot

python資料處理——數值轉one-hot

技術標籤:做個人吧數學pythonnumpy

_train_labels = [0,1,2,3,4,5,6,7,8,9,10]
_train_labels = np.array(_train_labels)
train_labels = np.zeros((_train_labels.shape[0],no_of_different_labels))
train_labels[np.arange(_train_labels.shape[0]),_train_labels-1] = 1