Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add ensemble of GNINA models #38

Merged
merged 5 commits into from Jul 17, 2022
Merged

Add ensemble of GNINA models #38

merged 5 commits into from Jul 17, 2022

Conversation

RMeli
Copy link
Owner

@RMeli RMeli commented Jul 17, 2022

Description

This PR adds the GNINAModelEnsemble class, which allow to combine different pre-trained GNINA models into a single class. Related to #33 and #35.

This PR also adds functions to easily load an ensemble of pre-trained models using their names, and to perform inference from the CLI. For the time being, the CLI requires a list of protein-ligand pairs in a file, instead of the protein and ligand directly as in GNINA.

The class assumes that the models perform both pose prediction and binding affinity prediction (which is the case for the pre-trained GNINA models, hence the GNINA name). A less strict class to be used with custom models (pose prediction only, or pose and flexible residues) will follow.

The dense model is not yet available since it appears to be problematic and it is under investigation. This means that the GNINA default model is not yet available.

PR Checklist

  • Tests
  • Documentation

@RMeli RMeli self-assigned this Jul 17, 2022
@codecov
Copy link

codecov bot commented Jul 17, 2022

Codecov Report

Merging #38 (1e9038d) into main (90fa1a7) will decrease coverage by 0.02%.
The diff coverage is 95.45%.

@RMeli RMeli merged commit 2e19e4d into main Jul 17, 2022
@RMeli RMeli deleted the ensemble branch July 17, 2022 13:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant