Skip to content

Update AlphaNet - #255

Merged
janosh merged 19 commits into
janosh:mainfrom
zmyybc:alphanet
Jun 16, 2025
Merged

Update AlphaNet#255
janosh merged 19 commits into
janosh:mainfrom
zmyybc:alphanet

Conversation

@zmyybc

@zmyybc zmyybc commented May 13, 2025

Copy link
Copy Markdown
Contributor

Description

Hi!
I am submitting 2 v1 AlphaNet models:

  • AlphaNet-v1-MPtrj: compiled model trained on MPtrj
  • AlphaNet-v1-OMA: Pretrained on OMAT24, finetuned on sALEX+MPtrj

Appreciate your work again!

Checklist

Please check the following items before submitting your PR:

  • I created a new folder and YAML metadata file models/<arch_name>/<model_variant>.yml for my submission. arch_name is the name of the architecture and model_variant.yml includes things like author details, training set names and important hyperparameters.
  • I added the my new model as a new attribute on the Model.<arch_name> enum in enums.py.
  • I uploaded the energy/force/stress model prediction file for the WBM test set to Figshare or another cloud storage service (<yyyy-mm-dd>-<model_variant>-preds.csv.gz).
  • I uploaded the model-relaxed structures file to Figshare or another cloud storage service in JSON lines format (<yyyy-mm-dd>-wbm-IS2RE-FIRE.jsonl.gz). JSON Lines allows fast loading of small numbers of structures with pandas.read_json(lines=True, nrows=100) for inspection.
  • I uploaded the phonon predictions to Figshare or another cloud storage service (<yyyy-mm-dd>-kappa-103-FIRE-<values-of-dist|fmax|symprec>.gz).
  • I included the urls to the Figshare files in the YAML metadata file (models/<arch_name>/<model_variant>.yml). If not using Figshare I have included the urls to the cloud storage service in the description of the PR.
  • I included the test script (test_<arch_name>_<task>.py for task in discovery, kappa, diatomics) that generated the prediction files.

Additional Information (Optional)

  • I included a training script (train_<arch_name>.py) if I trained a model specifically for this benchmark.
  • I included a readme.md with additional details about my model.

@janosh

janosh commented May 17, 2025

Copy link
Copy Markdown
Owner

thanks @zmyybc! i'll try to take a look tomorrow

@janosh janosh added the new model Model submission label May 17, 2025
@zmyybc

zmyybc commented May 22, 2025

Copy link
Copy Markdown
Contributor Author

ok

janosh added 5 commits June 16, 2025 08:18
- Corrected file paths in `enums.py` for `alphanet_mptrj` and `alphanet_oma`.
- Updated `pred_file` paths in `alphanet-mptrj.yml` and `alphanet-oma.yml` to reflect new directory structure.

@janosh janosh left a comment

Copy link
Copy Markdown
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

thanks a lot for this submission @zmyybc! 👍

apologies for the long wait for a review. nothing to do with the model, just bad time management on my part

@janosh
janosh merged commit dc9ef8f into janosh:main Jun 16, 2025
janosh added a commit that referenced this pull request Jul 31, 2026
* restore alphanet-mptrj.yml but set status: superseded

* add AlphaNet-v1-OMA metrics and reupload pred files to own figshare

- Corrected file paths in `enums.py` for `alphanet_mptrj` and `alphanet_oma`.
- Updated `pred_file` paths in `alphanet-mptrj.yml` and `alphanet-oma.yml` to reflect new directory structure.

* add e_form + e_hull energy parity plots and per-element e_hull errors

* fix test_df_each_(pred|err)

---------

Co-authored-by: Janosh Riebesell <janosh.riebesell@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

new model Model submission

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants