zrqiao / NeuralPLexer

NeuralPLexer: State-specific protein-ligand complex structure prediction with a multi-scale deep generative model
https://doi.org/10.1038/s42256-024-00792-z
BSD 3-Clause Clear License
256 stars 68 forks source link

`multi_pose_sampling` does not write `summary.csv` and does return per residue pLDDT scores #12

Open jsture opened 7 months ago

jsture commented 7 months ago

summary.csv is only generated by benchmarking functions and not the main predictor function. In addition, per residue scores are not returned.

One would probably have to add:

if confidence:
    plddt, plddt_lig = model.run_confidence_estimation(
        sample, output_struct, return_avg_stats=True
    )

    sample = model.run_confidence_estimation(
        sample, output_struct, return_avg_stats=False
    )

Then append that to some more lists and write those two as .csv files.

The logic and output of run_confidence_estimation is a bit too non-straightforward and undocumented for me to add this myself.

Samuel-gwb commented 7 months ago

Yes, it works fine just printing plddt and plddt_lig, while "sample" gives huge matrix setting return_avg_stats=False.

amorehead commented 5 months ago

I believe this PR addresses the missing plDDT values issue.