如何转换脚本的输出

时间:2019-02-21 21:34:26

标签: python

我正在创建一个Twitter Bot。我写了一个脚本,将生成文本。我正在尝试获取生成的文本,并通过另一个脚本发布它与生成的文本推文运行它。但是,我不知道如何获取字符串格式的generate-text-script的输出。

代码如下:

from textgenrnn import textgenrnn
t = textgenrnn('textgenrnn_weights.HDF5')
text = t.generate(1, temperature=.5)
print(text)

以下是Shell中的输出:

Shell Capture

2 个答案:

答案 0 :(得分:0)

从textgenrnn文档中,generate函数具有一个return_as_list参数,该参数默认为false,将结果打印到控制台。

如果您希望将结果作为字符串列表,请这样称呼:

output = t.generate(1, temperature=.5, return_as_list=True)
for string in output:
   print(string)

答案 1 :(得分:0)

您可以使用stderr重定向将所有初始信息发送为null,因此只剩下打印输出。

例如像python myscript.py 2>/dev/null

一样运行脚本

编辑:将其与有关使用return_as_list=True的其他答案结合起来,仅打印文本[0],我认为您已获得所需的输出。

例如:

from textgenrnn import textgenrnn
t = textgenrnn('reddit_rarepuppers_politics.hdf5')
text = t.generate(1, temperature=.5, return_as_list=True)
print(text[0])