TextGenRnn将生成的文本保存到文件

时间:2018-12-05 13:45:48

标签: python text neural-network recurrent-neural-network

我无法将textgenrnn生成的文本保存到文件中。

这是我的代码

from textgenrnn import textgenrnn
textgen = textgenrnn()
#after doing all the training work
textgen.generate_to_file('codes_new1.txt', 5)

我应该在当前文件夹的文件中得到5行。但这并没有发生,相反,我明白了。

TypeError: generate_to_file() takes 2 positional arguments but 3 were given

我继续检查了textgenrnn的源代码,它允许将**kwargs作为第三个参数。想知道怎么了。

1 个答案:

答案 0 :(得分:0)

解决了。添加了n=5而不是5,现在它可以正常工作了