cnn with tf.layers - 可以从tf.estimator.EstimatorSpec对象中获取预测的类标签吗？

Question

关注Guide to TF Layers: Building a Convolutional Neural Network 我可以看到训练和评估步骤正常工作，但我没有看到如何获得一个未标注数据集的预测类标签。我试过了：

# try predict
test_input_fn = tf.estimator.inputs.numpy_input_fn(
  x={"x": my_test_data},
  num_epochs=1,
  shuffle=False)
test_results = mnist_classifier.predict(input_fn=test_input_fn)
print(test_results)

这会产生一个“生成器对象Estimator.predict，位于0x00000000103DE990”，我无法操纵它来提取预测。

是否可以从此对象获取预测标签？如果不是，如何调整此cnn_model_fn以使其返回新的未标记数据集的预测标签？

Answer 1

我刚刚找到答案。我将其发布在cas中可以帮助其他人：

for el in test_results:
    print(el)

这将打印测试数据集中每个元素的标签和可能性。