如何从张量获得预测输出

Question

目前，我已经完成了 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 的概率最高。

Answer 1

使用argmax()方法获取张量各轴上最大值的索引。b = [0, 1, 2, 100, 3, 4, 5]print(np.argmax(b))3Tensorflow Argmax 文档Numpy Argmax 文档