Since I'm using your code, I think it's only fair to give something back.
I basically refactored the whole code base, since there were tons of stuff to improve.
There is no change in the core functionality, the results are identical.
Modifications:
Proper project structure
Added gitignore
Added Apache 2.0 License, since this is used in the metrics.py
Add poetry support and requirements.txt
Added main.py for users with examples how to use the code for single cases and batch mode
Proper docstrings
Tons of variable naming improvements, a good name is better than a comment
Since I'm using your code, I think it's only fair to give something back.
I basically refactored the whole code base, since there were tons of stuff to improve. There is no change in the core functionality, the results are identical.
Modifications: