Kohulan / Smiles-TO-iUpac-Translator

Transformer based SMILES to IUPAC Translator
MIT License
137 stars 27 forks source link

Token-level confidence #22

Closed OBrink closed 8 months ago

OBrink commented 1 year ago

I have added the option to get token-level confidence values based on the output of the final layer of the Transformer decoder. I don't have access to the checkpoints I cannot test it right now.

__ Sorry for the failing test, I they will only pass once the new model is on Zenodo and the link is updated.

Kohulan commented 1 year ago

Sorry for the failing test, I they will only pass once the new model is on Zenodo and the link is updated.

I will update the checkpoint and models first and then merge this. Thanks a lot for doing this!