search

attilanagy234 / neural-punctuator

Complimentary code for our paper Automatic punctuation restoration with BERT models
MIT License
48 stars 7 forks source link

Checking model output #7

Closed UsamaI000 closed 3 years ago

UsamaI000 commented 3 years ago

Hi, I wanted to know if you have implemented to check the model output at test time. Because we had aligned labels according to tokens so model output would be according to that. How can we convert them back to original which was before aligning ?

Best Regards

bana513 commented 3 years ago

Hi, here is how we did the conversion from predictions to text again: commit