mikahama / natas

Python 3 library for processing historical English
Apache License 2.0
64 stars 11 forks source link

new functionality to capture prediction scores from the OpenNMT model #7

Closed t-shoemaker closed 3 years ago

t-shoemaker commented 3 years ago

This PR adds a new argument to normalize_words() to optionally return scores for every candidate word outputted by the model. These scores are outputted by OpenNMT's Translator, so the changes only involve capturing this output.

Summary of changes

mikahama commented 3 years ago

I will merge this for now, but I still need to check the compatibility with my other libraries (UralicNLP and murre) before making a new release