Conversation
|
Hey @vue1999, thanks for the PR and its looking super good. A few things:
Let me know if you've got any ideas or are unsure about any of my suggestions, thanks! |
ml_peg/analysis/surfaces/cleavage_energy/analyse_cleavage_energy.py
Outdated
Show resolved
Hide resolved
ml_peg/analysis/surfaces/cleavage_energy/analyse_cleavage_energy.py
Outdated
Show resolved
Hide resolved
ml_peg/analysis/surfaces/cleavage_energy/analyse_cleavage_energy.py
Outdated
Show resolved
Hide resolved
ml_peg/analysis/surfaces/cleavage_energy/analyse_cleavage_energy.py
Outdated
Show resolved
Hide resolved
ml_peg/analysis/surfaces/cleavage_energy/analyse_cleavage_energy.py
Outdated
Show resolved
Hide resolved
ml_peg/analysis/surfaces/cleavage_energy/analyse_cleavage_energy.py
Outdated
Show resolved
Hide resolved
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
…gy.py Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
|
Thanks for the review @joehart2001! Happy to accept all suggestions and fine to keep just MAE for now. |
Co-authored-by: Joseph Hart <92541539+joehart2001@users.noreply.github.com>
There was a problem hiding this comment.
Thanks for this, @vue1999, and sorry for the delayed review on my side, it's looking great so far!
I've left a few minor comments. Could you run the pre-commit (see: https://ddmms.github.io/ml-peg/developer_guide/get_started.html#automatic-coding-style-check), to tidy up a few things for the automated tests?
I am also a bit concerned by the 30,000 files. The analysis just for a single model took me three minutes. We definitely still want access to the structure files, and it's good to save the data with them, but I wonder if it's worth also saving this to a simpler file format e.g. as you had previously with JSON (sorry if this feels back-and-forth) to be more practical.
It could also be worth exploring saving to a reduced number of files e.g. all structs in each mpid_dir, but I'm not sure it would make a huge difference.
Thoughts @joehart2001?
| idx = 0 | ||
| for mpid_dir in tqdm(sorted(d for d in data_dir.iterdir() if d.is_dir())): | ||
| for xyz_file in sorted(mpid_dir.glob("*.xyz")): | ||
| structs = read(xyz_file, index=":") |
There was a problem hiding this comment.
Could you set the charge and spin (multiplicity) for these structures? Orb-omol requires them to be set
| from ml_peg.analysis.utils.utils import ( | ||
| load_metrics_config, | ||
| mae, | ||
| write_density_trajectories, |
There was a problem hiding this comment.
I get an import error trying from this import. This branch may need rebasing?
| model_ref = [] | ||
|
|
||
| for xyz_file in sorted(model_dir.glob("*.xyz"), key=lambda p: int(p.stem)): | ||
| slab = read(xyz_file) |
There was a problem hiding this comment.
I don't think read is currently imported?
Pre-review checklist for PR author
PR author must check the checkboxes below when creating the PR.
Summary
Adding a benchmark to evaluate the accuracy of predicting cleavage energies of crystalline surfaces.
Linked issue
Resolves #403
Progress
Testing
MACE omat, ORB v3
New decorators/callbacks
no