目前，我已经从TensorFlow完成了我的功能神经元模型，我想将预测结果输出到CSV文件。
但是，当我得到预测时，它是一个具有8类输出概率的张量，即428行。
0 1 2 3 4 5 6 7
0 0.001258 0.971390 0.000170 0.005542 0.000077 0.002042 0.001015 0.018505
1 0.000663 0.989145 0.000010 0.000393 0.000642 0.000798 0.003435 0.004914
2 0.001139 0.939257 0.000232 0.017595 0.000112 0.000762 0.000535 0.040368
3 0.002857 0.970207 0.000041 0.000680 0.003005 0.000273 0.012651 0.010286
4 0.000706 0.992306 0.000017 0.000533 0.000908 0.000095 0.000808 0.004627
.
.
428
如何获取每个班级编号的最高概率作为输出？例如，第一个数据点为0，显然类别1具有最高的可能性。
答案 0 :(得分：1)
使用argmax()方法获取张量轴上最大的索引。
b = [0, 1, 2, 100, 3, 4, 5]
print(np.argmax(b))
3