В документации упоминается только использование командной строки для запуска сценария 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
Я не могу найти документацию о том, как это сделать. Пожалуйста, помогите!