polaris-hub / polaris

Foster the development of impactful AI models in drug discovery.
https://polaris-hub.github.io/polaris/
Apache License 2.0
94 stars 6 forks source link

Add ligand RMSD metrics #199

Closed zhu0619 closed 1 month ago

zhu0619 commented 2 months ago

Changelogs


Checklist:


The rmsd metrics are implemented based on the Posebuster codebase. A few modifications were made, replacing RDKit callables with equivalent Datamol functions. The RMSD calculation was also simplified to evaluate only the first conformer of the predicted molecule and to consider only heavy atoms.

Note: Posebusters provides a series of checkers, known as 'Posebuster Checkers,' to further filter out unwanted docked ligand conformers. To avoid over-complicating Polaris metrics module with case-specific metrics, users should apply these filters before uploading results to the Polaris hub.