如何从张量中获取预测输出答案

【问题标题】：How to get prediction output from the tensor如何从张量中获取预测输出
【发布时间】：2020-10-05 15:40:05
【问题描述】：

目前，我已经从 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 具有最高的概率。

【问题讨论】：

标签： python tensorflow

【解决方案1】：

使用argmax() 方法获取张量轴上最大值的索引。

b = [0, 1, 2, 100, 3, 4, 5]
print(np.argmax(b))

3

Tensorflow Argmax Doc

Numpy Argmax Doc

【讨论】：

谢谢。我使用tf.math.argmax(prediction,axis=1) 找到了对应于最高概率的索引。