Tensorflow标签预测给出InvalidArgumentError

Question

我正在尝试使用Tensorflow和Python预测标签。这是我目前经历的代码：

现在，问题来了。我无法从中获得输出，因为从此命令中会出现如下错误：

这是我用于预测的数据输入：

有什么想法我想念什么？

Answer 1

看起来您正在将“ argmax”的输出提供给下一个阶段。由“ argmax”返回的“ int的整数数组”是“沿轴的最大值的索引”。下一阶段将期待“浮动”。您可能希望索引所指向的内容进入数组，而不是索引本身。