one hot編碼(pytorch實現)

2022-09-17 16:03:16 字數 730 閱讀 8131

n = 5  #類別數

indices = torch.randint(0, n, size=(15,15)) #生成陣列元素0~5的二維陣列(15*15)

one_hot = torch.nn.functional.one_hot(indices, n) #size=(15, 15, n)

1.one-hot編碼(一維陣列、二維影象都可以):label = torch.nn.functional.one_hot(label, n)。 #

一維陣列的one hot編碼,n為類別,label為陣列

ps. (1)把陣列(m,n)轉換成(a,b,c),reshape/view時是將前者逐行讀取,轉換成後者的。

(2)還會補充one-hot編碼轉換成單通道影象的方法。

2.  one-hot編碼---label

對於一維陣列,results = one_hot_label.argmax(dim=1, keepdim=true)

或者a = [np.argmax(l)for lin one_hot]#將onehot編碼轉成一般編碼

Pytorch框架之one hot編碼函式

在乙個給定的向量中,按照設定的最值 可以是向量中包含的最大值 作為最高分類數 有也可以是自定義的最大值,設計one hot編碼的長度 最大值 1 詳見舉的例子吧 然後按照最大值建立乙個1 最大值 1 的維度大小的全零零向量 0,0,0,共最大值 1對應的個數 接著按照向量中的值,從第0位開始索引,將...

one hot編碼理解

one hot是比較常用的文字特徵特徵提取的方法。one hot編碼,又稱 獨熱編碼 其實就是用n位狀態暫存器編碼n個狀態,每個狀態都有獨立的暫存器位,且這些暫存器位中只有一位有效,說白了就是只能有乙個狀態。下面舉例說明 有四個樣本,每個樣本有三種特徵 feature1 feature2 featu...

onehot編碼解釋

one hot編碼,又稱為一位有效編碼,主要是採用n位狀態暫存器來對n個狀態進行編碼,每個狀態都由他獨立的暫存器位,並且在任意時候只有一位有效。one hot編碼是分類變數作為二進位制向量的表示。這首先要求將分類值對映到整數值。然後,每個整數值被表示為二進位制向量,除了整數的索引之外,它都是零值,它...