如何运行OpenNMT-py模型以在jupyter笔记本中翻译给定字符串?

时间:2019-03-26 23:48:07

标签: python jupyter-notebook pytorch opennmt

文档仅提及使用命令行来运行translate.py存储库中已给出的OpenNMT-py脚本,即使用

python translate.py -model model_name.pt -src data/src-test.txt -output pred.txt

很好,但是如何使用onmt库将经过训练的模型加载到jupyter笔记本中并随时进行翻译?我希望它能够在jupyter中执行以下操作:

model = onmt.module.load(/path/to/model.pt)
text = "hola como estas"
model.translate(text)

Output:
>>>Hello how are you

我找不到有关如何执行此操作的任何文档。请帮忙!

0 个答案:

没有答案