Tensorflow:classifier.predict和predict_classes

时间:2017-10-24 17:02:12

标签: tensorflow classification

系统信息

  • 自定义代码:不,它是https://www.tensorflow.org/get_started/estimator
  • 中的代码
  • system:Apple
  • 操作系统:Mac OsX 10.13
  • TensorFlow版本:1.3.0
  • Python版本:3.6.3
  • GPU型号:AMD FirePro D700(实际上是两个这样的GPU)

描述问题

亲爱的, 我正在运行简单的虹膜程序: 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 个答案:

答案 0 :(得分:0)

这看起来像一个numpy问题。 brand <- sub("(\\w+).*", "\\1", dat$brand) 是numpy代表字符串array([b'1'], dtype=object)的一种方式。