如何从张量获得预测输出

Question

目前，我已经完成了 TensorFlow 的功能神经元模型，我想将预测输出到 CSV 文件。然而，当我得到预测时，它是一个具有 8 类输出概率的张量，428 行。        0           1           2           3           4          5            6          70   0.001258    0.971390    0.000170    0.005542    0.000077    0.002042    0.001015    0.0185051   0.000663    0.989145    0.000010    0.000393    0.000642    0.000798    0.003435    0.0049142   0.001139    0.939257    0.000232    0.017595    0.000112    0.000762    0.000535    0.0403683   0.002857    0.970207    0.000041    0.000680    0.003005    0.000273    0.012651    0.0102864   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))

3

Tensorflow Argmax 文档

Numpy Argmax 文档