张量流：classifier.predict 和 predict_classes

Posted 2023-03-13

【中文标题】张量流：classifier.predict 和 predict_classes

【英文标题】：Tensorflow: classifier.predict and predicted_classes

【发布时间】：2018-04-05 14:01:20

【问题描述】：

系统信息

自定义代码：不，它是https://www.tensorflow.org/get_started/estimator中的那个 系统：苹果 操作系统：Mac OsX 10.13 TensorFlow 版本：1.3.0 Python 版本：3.6.3 GPU 型号：AMD FirePro D700（实际上是两个这样的 GPU）

描述问题

亲爱的， 我正在运行简单的 iris 程序： https://www.tensorflow.org/get_started/estimator 在 python 3.6.3 和 tensorflow 1.3.0 下。 该程序正确执行，除了最后一部分，即与混淆矩阵相关的部分。 事实上，我得到的混淆矩阵的结果是： 新样本，类别预测：[array([b'1'], dtype=object), array([b'2'], dtype=object)] 而不是预期的输出： 新样本，类别预测：[1 2] 最新版本中混淆矩阵有什么变化吗？ 如果是这样，我应该如何修改那部分代码？ 非常感谢您的帮助！ 最好的祝福 伊万

源代码/日志

https://www.tensorflow.org/get_started/estimator

【参考方案1】：

这看起来像一个 numpy 问题。 array([b'1'], dtype=object) 是 numpy 表示字符串 '1' 的一种方式。