我正在努力 文件分类问题 从Kaggle。
它有5个班级 - 商业"、"科技"、"政治"、"体育"、"娱乐
我已经训练了我的深度学习模型，也得到了测试集的结果。但我得到的结果是不同类的概率列表。
如何从我得到的输出中获取实际的类（标签）？
我的神经网络架构是这样的
你应该选择数值最高的条目作为预测类。例如，在你提供的例子中，你应该选择最高值的条目作为预测类。[0.045, 0.030, 0.015, 0.889, 0.019]，预测的类是概率值最高的第四个类（即.., idx=3)，它的概率值最高。
在你提供的例子中： ，预测的类是概率值最高的第四类（即， ）。氩气 NumPy的函数可能是你应该使用的。考虑到 pred 是你的网络的输出概率，形状为。(batch_size, num_labels)，那么 np.argmax(pred, axis=1) 将为您提供与预测类相关的指数（即标签）。