幾行python代碼實現one-hot編碼

#假設y是np.array數組形式的數組,形狀爲(row, col)
#t是類別標籤,也爲數組,形狀爲(row,),比如[0,2,3,5,2,6]
one_hot_t= np.zeros_like(y)  #生成和y形狀一樣的元素爲零的數組
for j, i in zip(range(t.size), t):
    #有多少個樣本就應該對應多少個標籤
    one_hot_t[j][i] = 1      #變爲one-hot類型標籤:j表示樣本,i表示標籤索引

one_hot_t                   #獲得one-hot編碼

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章