张量流如何进行打印预测

2024-04-29 • 问答

我建立了一个简单的预测模型（我必须预测“目标”，它可以是0或1），并且希望以以下格式打印并保存到文件数据中：

$compile

你知道怎么做吗？现在的打印值是：

id_of_prediction,prediction
0,1
1,0
2,1
3,1
...

我的代码：

<generator object Estimator.predict at 0x7f718d5621a8>

您拥有的对象是生成器。目的是要迭代地返回数据，您可以在此处阅读有关以下内容的更多信息：https://wiki.python.org/moin/Generators

要展开生成器，可以将其传递给list函数。它占用了生成器的每个“元素”，并将其逐个放入新列表中。

因此，您可以做的是展开生成器，预览其内容并将其转储为CSV或类似格式。

您可以通过以下方式打印预测：

print(next(predictions))

while(True):
    try:
        print(next(predictions))
    except:
        break

next()方法在python中用于迭代生成器。