在 Python 中将一个热编码结果转换回单列

Posted 2023-03-12

【中文标题】在 Python 中将一个热编码结果转换回单列

【英文标题】：Coverting Back One Hot Encoded Results back to single Column in Python

【发布时间】：2018-01-15 01:25:06

【问题描述】：

我正在使用 Keras 进行多类分类。它包含 5 类输出。我使用一种热编码将单个类向量转换为矩阵并制作了一个模型。现在要评估模型，我想将 5 类概率结果转换回单列。

我将其作为 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]

【参考方案1】：

你好像在找np.argmax:

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

【讨论】：

打算把我的评论变成答案。打败我。 +1