隱藏回一個熱編碼結果返回到Python中的單列

我正在使用Keras進行多類分類。它包含5類輸出。我使用一種熱門編碼將單類矢量轉換爲矩陣並製作了一個模型。現在評估模型，我想將5類概率結果轉換回單列。隱藏回一個熱編碼結果返回到Python中的單列

我得到這個在numpy的陣列格式輸出

..................0..................1............................2.......................3.............................4 

5.35433665e-02 1.72592481e-05 1.49291719e-03 9.44392741e-01 
    5.53713820e-04 
    1.97096306e-05 2.08907949e-08 3.11666554e-07 1.40611945e-07 
    9.99979794e-01 
    9.99999225e-01 2.42999278e-07 1.58917388e-07 7.84497018e-08 
    2.85837785e-07 
    7.09977685e-05 1.02068476e-09 1.38186664e-07 9.99928594e-01 
    2.73126261e-07 
    1.29937407e-05 2.49388819e-07 9.99986231e-01 4.76015231e-07 
    7.39421040e-08

想這個矩陣轉換爲

[3,4,0,3,2]

來源

2017-07-19 Abhik Sarkar

好像你正在尋找np.argmax：

import numpy as np 
class_labels = np.argmax(class_prob, axis=1) # assuming you have n-by-5 class_prob

來源

2017-07-19 07:12:44 Shai

當時將我的評論變成答案。打敗我吧。 +1 –

隱藏回一個熱編碼結果返回到Python中的單列

回答

相關問題