我正在创建一个Twitter Bot。我写了一个脚本,将生成文本。我正在尝试获取生成的文本,并通过另一个脚本发布它与生成的文本推文运行它。但是,我不知道如何获取字符串格式的generate-text-script的输出。
代码如下:
from textgenrnn import textgenrnn
t = textgenrnn('textgenrnn_weights.HDF5')
text = t.generate(1, temperature=.5)
print(text)
以下是Shell中的输出:
答案 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])