gcnwm commited on
Commit
8775fd6
·
1 Parent(s): bb32642

Realign the curated model repository with the v4 workspace snapshot

Browse files

Constraint: The curated model repo must stay on main while the source snapshot lives under results/al_v4
Rejected: Keep the stale MHC-inclusive publication | it no longer matched the v4 selection
Confidence: medium
Scope-risk: moderate
Directive: Rebuild the NequIP artifact binaries from v4 checkpoints if those checkpoints are later restored
Tested: git diff --check; JSON parse sanity; refreshed MACE LAMMPS exports from v4 raw files
Not-tested: Fresh NequIP binary regeneration from v4 checkpoints, which are not present in the local results tree

README.md CHANGED
@@ -1,63 +1,22 @@
1
- ---
2
- library_name: pytorch
3
- tags:
4
- - mace
5
- - nequip
6
- - active-learning
7
- - interatomic-potentials
8
- - atomistic-machine-learning
9
- - computational-chemistry
10
- - ase
11
- - lammps
12
- ---
13
- # FYP raw and compiled active-learning checkpoints for rMD17 ethanol
14
 
15
- This repository contains fresh model publication artifacts from the bachelor-thesis
16
- active-learning study on **rMD17 ethanol**. Each bundle corresponds to the
17
- best final-iteration seed by **forces MAE** for one architecture/strategy family.
 
18
 
19
- The publication was rebuilt from scratch from the local `results/al` tree rather than
20
- reusing previously uploaded assets.
21
 
22
- ## Included artifact types
 
 
 
 
 
23
 
24
- - **Raw MACE checkpoints**: original `.model` files copied from the final selected iteration.
25
- - **Compiled MACE checkpoints**: official LAMMPS exports produced with
26
- `python -m mace.cli.create_lammps_model --format libtorch`.
27
- - **Raw NequIP checkpoints**: original Lightning `.ckpt` files copied from the selected iteration.
28
- - **Packaged NequIP checkpoints**: official `.nequip.zip` packages produced with `nequip-package build`.
29
- - **Compiled NequIP checkpoints**: official AOTInductor artifacts produced with
30
- `nequip-compile --mode aotinductor --device cuda --target ase`.
31
-
32
- ## Selected bundle summary
33
-
34
- | Bundle | Architecture | Strategy | Best seed | Final iteration | Forces MAE (meV/Å) | Energy MAE (meV) | Raw files | Packaged files | Compiled files |
35
- | --- | --- | --- | ---: | ---: | ---: | ---: | ---: | ---: | ---: |
36
- | mace-random | MACE | RANDOM | 2 | 9 | 9.20 | 1.71 | 1 | 0 | 1 |
37
- | mace-qbc | MACE | QBC | 2 | 9 | 8.51 | 1.60 | 4 | 0 | 4 |
38
- | mace-mhc | MACE | MHC | 2 | 9 | 9.97 | 2.07 | 1 | 0 | 4 |
39
- | nequip-random | NequIP | RANDOM | 3 | 9 | 8.89 | 4.74 | 1 | 1 | 1 |
40
- | nequip-qbc | NequIP | QBC | 1 | 9 | 7.57 | 1.48 | 3 | 3 | 3 |
41
-
42
- ## Important details
43
 
44
  - `mace-qbc/` contains the full four-member committee for the selected QBC seed.
45
- - `mace-mhc/compiled/` contains one compiled LAMMPS export per available model head.
46
  - `nequip-qbc/` contains one raw/package/compiled triplet per selected committee member.
47
- - The NequIP training checkpoints embed legacy absolute workspace paths, so packaging was
48
- performed from a temporary local mirror with rewritten checkpoint-path prefixes before
49
- running the official NequIP CLI.
50
- - The compiled NequIP `.nequip.pt2` artifacts are **device-specific** AOTInductor exports
51
- built for `cuda` on this machine; the `.nequip.zip` packages are the more portable
52
- redistribution artifact.
53
-
54
- ## Provenance
55
-
56
- - Hugging Face repo: `https://huggingface.co/gcnwm/fyp-active-learning-models`
57
- - Source thesis workspace commit: `8fea4f53dcdf581ca23a820ab1cc8de9708f7aac`
58
- - Companion production-code repository: `https://github.com/gcnwm/fyp-production`
59
- - Companion production-code commit: `433f9ee3872a84cfeabf19a5d64fca9277d5cf09`
60
- ## Citation
61
-
62
- If you use these checkpoints, please cite the accompanying thesis project and the
63
- upstream **MACE**, **NequIP**, and **rMD17** papers.
 
1
+ # Curated FYP Active-Learning Checkpoints
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
+ This Hugging Face model repository stores a curated export of the strongest
4
+ final-iteration checkpoints from the MACE and NequIP active-learning study on
5
+ rMD17 ethanol. Each bundle is selected by the best final forces MAE across the
6
+ three repeated experiment seeds for its architecture/strategy family.
7
 
8
+ ## Included bundles
 
9
 
10
+ | Bundle | Architecture | Strategy | Selected seed | Final forces MAE (meV/Å) | Final energy MAE (meV) | Files |
11
+ | --- | --- | --- | ---: | ---: | ---: | ---: |
12
+ | mace-random | MACE | RANDOM | 1 | 9.45 | 1.79 | 2 |
13
+ | mace-qbc | MACE | QBC | 3 | 8.61 | 1.69 | 8 |
14
+ | nequip-random | NequIP | RANDOM | 2 | 10.66 | 4.24 | 7 |
15
+ | nequip-qbc | NequIP | QBC | 3 | 9.53 | 6.52 | 21 |
16
 
17
+ ## Notes
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
 
19
  - `mace-qbc/` contains the full four-member committee for the selected QBC seed.
 
20
  - `nequip-qbc/` contains one raw/package/compiled triplet per selected committee member.
21
+ - Per-bundle `metadata.json` files record the selected seed, final metrics, and
22
+ original relative source paths from the thesis workspace.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
mace-mhc/compiled/mace_mhc_4h_stagetwo-head0.model-lammps.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ef2ef5bc7d39e662341c06d40f5cf2562bd4ebc3fab9a0659de7baa6cfed2b1e
3
- size 7737953
 
 
 
 
mace-mhc/compiled/mace_mhc_4h_stagetwo-head1.model-lammps.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:b80edbb41eba2f9173642837ca7af1571dc7d77a4433ddd98288f8bdf562d21c
3
- size 7737953
 
 
 
 
mace-mhc/compiled/mace_mhc_4h_stagetwo-head2.model-lammps.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:4884c7cd61f90551206019a59cba9dd832fae0185a6caae35af6dd9444dab5ff
3
- size 7737953
 
 
 
 
mace-mhc/compiled/mace_mhc_4h_stagetwo-head3.model-lammps.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:d3978bf52d35e94d1b6cae79ea27a9bbe5aea4f82fb6228ef45d79e18da9ad02
3
- size 7737953
 
 
 
 
mace-mhc/metadata.json DELETED
@@ -1,59 +0,0 @@
1
- {
2
- "architecture": "MACE",
3
- "bundle": "mace-mhc",
4
- "compiled_files": [
5
- {
6
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head0 --format libtorch",
7
- "format": "libtorch",
8
- "head": "head0",
9
- "published_file": "mace_mhc_4h_stagetwo-head0.model-lammps.pt",
10
- "raw_file": "mace_mhc_4h_stagetwo.model"
11
- },
12
- {
13
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head1 --format libtorch",
14
- "format": "libtorch",
15
- "head": "head1",
16
- "published_file": "mace_mhc_4h_stagetwo-head1.model-lammps.pt",
17
- "raw_file": "mace_mhc_4h_stagetwo.model"
18
- },
19
- {
20
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head2 --format libtorch",
21
- "format": "libtorch",
22
- "head": "head2",
23
- "published_file": "mace_mhc_4h_stagetwo-head2.model-lammps.pt",
24
- "raw_file": "mace_mhc_4h_stagetwo.model"
25
- },
26
- {
27
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head3 --format libtorch",
28
- "format": "libtorch",
29
- "head": "head3",
30
- "published_file": "mace_mhc_4h_stagetwo-head3.model-lammps.pt",
31
- "raw_file": "mace_mhc_4h_stagetwo.model"
32
- }
33
- ],
34
- "final_iteration": 9,
35
- "final_metrics": {
36
- "energy_mae": 0.00207461045447144,
37
- "energy_rmse": 0.002818075833732281,
38
- "forces_mae": 0.00997466210127421,
39
- "forces_rmse": 0.014573431573543202,
40
- "iteration": 9,
41
- "max_disagreement": 0.0,
42
- "mean_disagreement": 0.0,
43
- "n_labeled": 550,
44
- "training_time": 2519.816365480423
45
- },
46
- "packaged_files": [],
47
- "raw_files": [
48
- {
49
- "published_file": "mace_mhc_4h_stagetwo.model",
50
- "source_file": "results/al/mace_mhc_seed2/iter09_committee/models/mace_mhc_4h_stagetwo.model"
51
- }
52
- ],
53
- "selected_seed": 2,
54
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
55
- "source_files": [
56
- "results/al/mace_mhc_seed2/iter09_committee/models/mace_mhc_4h_stagetwo.model"
57
- ],
58
- "strategy": "mhc"
59
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
mace-mhc/raw/mace_mhc_4h_stagetwo.model DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:1ba54febe19dc369f833513bbfa5cee4d7ca6374fc79caa66b05c958a4fd41fa
3
- size 7639289
 
 
 
 
mace-qbc/compiled/mace_qbc_0_stagetwo.model-lammps.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fd39686abad1c903e8b5b1c6af53003f911f811553bc37e2075ce34ab534492e
3
  size 7722036
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42d642e82691b6b50d2bf26842a6427ea8b3f1c608af072472f92cacff1497c1
3
  size 7722036
mace-qbc/compiled/mace_qbc_1_stagetwo.model-lammps.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9baca252ede471cfafe2227d1527d167fdac7218344f122d64aa04936b0ca560
3
  size 7722036
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:01cfb69fec7746f5c2fe56811d02592fdfc816d4706bced95a0ab9e9817ed3fa
3
  size 7722036
mace-qbc/compiled/mace_qbc_2_stagetwo.model-lammps.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5475a3bffb2cbba659cbc87c8ce8a23c2cdfb59b89ce8a7b8ba96c25ee20a531
3
  size 7722036
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b25b1c50a87a3e3e07dd4abe04477f9e9a07205fe906d5e430ecfec37309e6f6
3
  size 7722036
mace-qbc/compiled/mace_qbc_3_stagetwo.model-lammps.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:126bfb03f77ba9f20cf5d609ee11f314d628b8d4457042fcb375ec5ae71eb0b6
3
  size 7722036
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d48f7209a223d052289e0255a05520b2d1de058136ab45f3fa2206456cb0b127
3
  size 7722036
mace-qbc/metadata.json CHANGED
@@ -33,17 +33,34 @@
33
  ],
34
  "final_iteration": 9,
35
  "final_metrics": {
36
- "energy_mae": 0.0015974740571928123,
37
- "energy_rmse": 0.002195391255950762,
38
- "forces_mae": 0.008513434624789339,
39
- "forces_rmse": 0.012237725244050215,
 
 
 
 
 
 
 
40
  "iteration": 9,
 
41
  "max_disagreement": 0.0,
42
  "mean_disagreement": 0.0,
43
  "n_labeled": 550,
44
- "training_time": 7162.63129401207
 
 
 
45
  },
46
  "packaged_files": [],
 
 
 
 
 
 
47
  "raw_files": [
48
  {
49
  "published_file": "mace_qbc_0_stagetwo.model",
@@ -62,13 +79,13 @@
62
  "source_file": "results/al/mace_qbc_seed2/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
63
  }
64
  ],
65
- "selected_seed": 2,
66
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
67
  "source_files": [
68
- "results/al/mace_qbc_seed2/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model",
69
- "results/al/mace_qbc_seed2/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model",
70
- "results/al/mace_qbc_seed2/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model",
71
- "results/al/mace_qbc_seed2/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
72
  ],
73
  "strategy": "qbc"
74
  }
 
33
  ],
34
  "final_iteration": 9,
35
  "final_metrics": {
36
+ "acquisition_rule": "uncertainty",
37
+ "acquisition_seed": null,
38
+ "candidate_n": 0,
39
+ "candidate_policy": "proposal_stream_filtered",
40
+ "candidate_seed": null,
41
+ "energy_mae": 0.001692634182771144,
42
+ "energy_rmse": 0.0022133917348856097,
43
+ "evaluation_policy": "fixed_member",
44
+ "forces_mae": 0.008605168772635183,
45
+ "forces_rmse": 0.01183799714274852,
46
+ "is_acquisition_iteration": false,
47
  "iteration": 9,
48
+ "labeled_n_after_acquisition": 550,
49
  "max_disagreement": 0.0,
50
  "mean_disagreement": 0.0,
51
  "n_labeled": 550,
52
+ "selected_n": 0,
53
+ "train_n_before_acquisition": 550,
54
+ "training_policy": "scratch",
55
+ "training_time": 6118.126273155212
56
  },
57
  "packaged_files": [],
58
+ "published_files": [
59
+ "mace_qbc_0_stagetwo.model",
60
+ "mace_qbc_1_stagetwo.model",
61
+ "mace_qbc_2_stagetwo.model",
62
+ "mace_qbc_3_stagetwo.model"
63
+ ],
64
  "raw_files": [
65
  {
66
  "published_file": "mace_qbc_0_stagetwo.model",
 
79
  "source_file": "results/al/mace_qbc_seed2/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
80
  }
81
  ],
82
+ "selected_seed": 3,
83
+ "source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
84
  "source_files": [
85
+ "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model",
86
+ "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model",
87
+ "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model",
88
+ "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
89
  ],
90
  "strategy": "qbc"
91
  }
mace-qbc/raw/mace_qbc_0_stagetwo.model CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a1b97c64219b53f4d84ad36756a0e546d756b52a13a7dd28d17cbade7529c7a7
3
  size 7623775
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96b85694e2869521530dcb9b22835693ce34c3a81af2fc5911167205c1d5d70d
3
  size 7623775
mace-qbc/raw/mace_qbc_1_stagetwo.model CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:91151441c2f48900b6407d261820f58237451bb4695411553b6dae407c6b60f2
3
  size 7623775
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d39617aa283c96832ed0a8c8a55e3bc51b82e6a1059869278ba89fd6ac7014d1
3
  size 7623775
mace-qbc/raw/mace_qbc_2_stagetwo.model CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c114dc7a18b186c0f79b29fb65b1694ad4009b613c7a8736325c3ed685262b32
3
  size 7623775
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4bde3567e7e7b1a4e56a34954ca7bb2d76bbabfbcf104213c14225a63448f53a
3
  size 7623775
mace-qbc/raw/mace_qbc_3_stagetwo.model CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:130edb12fb652b4fcb996a95276d4b68695faa22379c94f61aeb94e1b84f7733
3
  size 7623775
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b206ba224196a6f90e59a1adaae677bb3cab9f1fc4e8355c0af3e8b081cff5a
3
  size 7623775
mace-random/compiled/mace_single_0_stagetwo.model-lammps.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed53038609433d5662c3b66b1bb3971b1332132a01186c322491df3f4e1f53f2
3
  size 7723131
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f1b479bbb6bcb7c7b5589796187729f73488e2540386865d258d0d35dc6d9dd
3
  size 7723131
mace-random/metadata.json CHANGED
@@ -12,27 +12,41 @@
12
  ],
13
  "final_iteration": 9,
14
  "final_metrics": {
15
- "energy_mae": 0.0017055419633352357,
16
- "energy_rmse": 0.0024813228812345487,
17
- "forces_mae": 0.009196984049307067,
18
- "forces_rmse": 0.013864793672371095,
 
 
 
 
 
 
 
19
  "iteration": 9,
 
20
  "max_disagreement": 0.0,
21
  "mean_disagreement": 0.0,
22
  "n_labeled": 550,
23
- "training_time": 1908.3744568824768
 
 
 
24
  },
25
  "packaged_files": [],
 
 
 
26
  "raw_files": [
27
  {
28
  "published_file": "mace_single_0_stagetwo.model",
29
  "source_file": "results/al/random_seed2/iter09_model/models/mace_single_0_stagetwo.model"
30
  }
31
  ],
32
- "selected_seed": 2,
33
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
34
  "source_files": [
35
- "results/al/random_seed2/iter09_model/models/mace_single_0_stagetwo.model"
36
  ],
37
  "strategy": "random"
38
  }
 
12
  ],
13
  "final_iteration": 9,
14
  "final_metrics": {
15
+ "acquisition_rule": "random",
16
+ "acquisition_seed": null,
17
+ "candidate_n": 0,
18
+ "candidate_policy": "proposal_stream_filtered",
19
+ "candidate_seed": null,
20
+ "energy_mae": 0.0017917274611536414,
21
+ "energy_rmse": 0.002695636120741299,
22
+ "evaluation_policy": "fixed_member",
23
+ "forces_mae": 0.009449580705448683,
24
+ "forces_rmse": 0.014669491447455139,
25
+ "is_acquisition_iteration": false,
26
  "iteration": 9,
27
+ "labeled_n_after_acquisition": 550,
28
  "max_disagreement": 0.0,
29
  "mean_disagreement": 0.0,
30
  "n_labeled": 550,
31
+ "selected_n": 0,
32
+ "train_n_before_acquisition": 550,
33
+ "training_policy": "scratch",
34
+ "training_time": 1608.996175289154
35
  },
36
  "packaged_files": [],
37
+ "published_files": [
38
+ "mace_single_0_stagetwo.model"
39
+ ],
40
  "raw_files": [
41
  {
42
  "published_file": "mace_single_0_stagetwo.model",
43
  "source_file": "results/al/random_seed2/iter09_model/models/mace_single_0_stagetwo.model"
44
  }
45
  ],
46
+ "selected_seed": 1,
47
+ "source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
48
  "source_files": [
49
+ "results/al_v4/v4_primary_fixed_member/mace_random_seed1/iter09_model/models/mace_single_0_stagetwo.model"
50
  ],
51
  "strategy": "random"
52
  }
mace-random/raw/mace_single_0_stagetwo.model CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4feca18a142d08e0f8dd03922aa2f2b7e5bcd86a38c0a411807d6689bb94bff4
3
  size 7624109
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:011f6433dff21de38318de0d8e86c25a84529a4e065d1ba038a9c65dcafb89d0
3
  size 7624109
model_index.json CHANGED
@@ -4,7 +4,7 @@
4
  "bundle": "mace-random",
5
  "compiled_files": [
6
  {
7
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-random/raw/mace_single_0_stagetwo.model --format libtorch",
8
  "format": "libtorch",
9
  "head": "Default",
10
  "published_file": "mace_single_0_stagetwo.model-lammps.pt",
@@ -13,27 +13,38 @@
13
  ],
14
  "final_iteration": 9,
15
  "final_metrics": {
16
- "energy_mae": 0.0017055419633352357,
17
- "energy_rmse": 0.0024813228812345487,
18
- "forces_mae": 0.009196984049307067,
19
- "forces_rmse": 0.013864793672371095,
 
 
 
 
 
 
 
20
  "iteration": 9,
 
21
  "max_disagreement": 0.0,
22
  "mean_disagreement": 0.0,
23
  "n_labeled": 550,
24
- "training_time": 1908.3744568824768
 
 
 
25
  },
26
  "packaged_files": [],
27
  "raw_files": [
28
  {
29
  "published_file": "mace_single_0_stagetwo.model",
30
- "source_file": "results/al/random_seed2/iter09_model/models/mace_single_0_stagetwo.model"
31
  }
32
  ],
33
- "selected_seed": 2,
34
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
35
  "source_files": [
36
- "results/al/random_seed2/iter09_model/models/mace_single_0_stagetwo.model"
37
  ],
38
  "strategy": "random"
39
  },
@@ -42,28 +53,28 @@
42
  "bundle": "mace-qbc",
43
  "compiled_files": [
44
  {
45
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-qbc/raw/mace_qbc_0_stagetwo.model --format libtorch",
46
  "format": "libtorch",
47
  "head": "Default",
48
  "published_file": "mace_qbc_0_stagetwo.model-lammps.pt",
49
  "raw_file": "mace_qbc_0_stagetwo.model"
50
  },
51
  {
52
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-qbc/raw/mace_qbc_1_stagetwo.model --format libtorch",
53
  "format": "libtorch",
54
  "head": "Default",
55
  "published_file": "mace_qbc_1_stagetwo.model-lammps.pt",
56
  "raw_file": "mace_qbc_1_stagetwo.model"
57
  },
58
  {
59
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-qbc/raw/mace_qbc_2_stagetwo.model --format libtorch",
60
  "format": "libtorch",
61
  "head": "Default",
62
  "published_file": "mace_qbc_2_stagetwo.model-lammps.pt",
63
  "raw_file": "mace_qbc_2_stagetwo.model"
64
  },
65
  {
66
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-qbc/raw/mace_qbc_3_stagetwo.model --format libtorch",
67
  "format": "libtorch",
68
  "head": "Default",
69
  "published_file": "mace_qbc_3_stagetwo.model-lammps.pt",
@@ -72,104 +83,56 @@
72
  ],
73
  "final_iteration": 9,
74
  "final_metrics": {
75
- "energy_mae": 0.0015974740571928123,
76
- "energy_rmse": 0.002195391255950762,
77
- "forces_mae": 0.008513434624789339,
78
- "forces_rmse": 0.012237725244050215,
 
 
 
 
 
 
 
79
  "iteration": 9,
 
80
  "max_disagreement": 0.0,
81
  "mean_disagreement": 0.0,
82
  "n_labeled": 550,
83
- "training_time": 7162.63129401207
 
 
 
84
  },
85
  "packaged_files": [],
86
  "raw_files": [
87
  {
88
  "published_file": "mace_qbc_0_stagetwo.model",
89
- "source_file": "results/al/mace_qbc_seed2/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model"
90
  },
91
  {
92
  "published_file": "mace_qbc_1_stagetwo.model",
93
- "source_file": "results/al/mace_qbc_seed2/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model"
94
  },
95
  {
96
  "published_file": "mace_qbc_2_stagetwo.model",
97
- "source_file": "results/al/mace_qbc_seed2/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model"
98
  },
99
  {
100
  "published_file": "mace_qbc_3_stagetwo.model",
101
- "source_file": "results/al/mace_qbc_seed2/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
102
  }
103
  ],
104
- "selected_seed": 2,
105
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
106
  "source_files": [
107
- "results/al/mace_qbc_seed2/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model",
108
- "results/al/mace_qbc_seed2/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model",
109
- "results/al/mace_qbc_seed2/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model",
110
- "results/al/mace_qbc_seed2/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
111
  ],
112
  "strategy": "qbc"
113
  },
114
- {
115
- "architecture": "MACE",
116
- "bundle": "mace-mhc",
117
- "compiled_files": [
118
- {
119
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head0 --format libtorch",
120
- "format": "libtorch",
121
- "head": "head0",
122
- "published_file": "mace_mhc_4h_stagetwo-head0.model-lammps.pt",
123
- "raw_file": "mace_mhc_4h_stagetwo.model"
124
- },
125
- {
126
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head1 --format libtorch",
127
- "format": "libtorch",
128
- "head": "head1",
129
- "published_file": "mace_mhc_4h_stagetwo-head1.model-lammps.pt",
130
- "raw_file": "mace_mhc_4h_stagetwo.model"
131
- },
132
- {
133
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head2 --format libtorch",
134
- "format": "libtorch",
135
- "head": "head2",
136
- "published_file": "mace_mhc_4h_stagetwo-head2.model-lammps.pt",
137
- "raw_file": "mace_mhc_4h_stagetwo.model"
138
- },
139
- {
140
- "command": "pixi run python -m mace.cli.create_lammps_model /tmp/fyp-hf-models-nao75_pz/publish/mace-mhc/raw/mace_mhc_4h_stagetwo.model --head head3 --format libtorch",
141
- "format": "libtorch",
142
- "head": "head3",
143
- "published_file": "mace_mhc_4h_stagetwo-head3.model-lammps.pt",
144
- "raw_file": "mace_mhc_4h_stagetwo.model"
145
- }
146
- ],
147
- "final_iteration": 9,
148
- "final_metrics": {
149
- "energy_mae": 0.00207461045447144,
150
- "energy_rmse": 0.002818075833732281,
151
- "forces_mae": 0.00997466210127421,
152
- "forces_rmse": 0.014573431573543202,
153
- "iteration": 9,
154
- "max_disagreement": 0.0,
155
- "mean_disagreement": 0.0,
156
- "n_labeled": 550,
157
- "training_time": 2519.816365480423
158
- },
159
- "packaged_files": [],
160
- "raw_files": [
161
- {
162
- "published_file": "mace_mhc_4h_stagetwo.model",
163
- "source_file": "results/al/mace_mhc_seed2/iter09_committee/models/mace_mhc_4h_stagetwo.model"
164
- }
165
- ],
166
- "selected_seed": 2,
167
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
168
- "source_files": [
169
- "results/al/mace_mhc_seed2/iter09_committee/models/mace_mhc_4h_stagetwo.model"
170
- ],
171
- "strategy": "mhc"
172
- },
173
  {
174
  "architecture": "NequIP",
175
  "bundle": "nequip-random",
@@ -184,15 +147,26 @@
184
  ],
185
  "final_iteration": 9,
186
  "final_metrics": {
187
- "energy_mae": 0.004738401647407954,
188
- "energy_rmse": 0.005157445425014896,
189
- "forces_mae": 0.008885047953261107,
190
- "forces_rmse": 0.013798360721764913,
 
 
 
 
 
 
 
191
  "iteration": 9,
 
192
  "max_disagreement": 0.0,
193
  "mean_disagreement": 0.0,
194
  "n_labeled": 550,
195
- "training_time": 2944.7004199028015
 
 
 
196
  },
197
  "packaged_files": [
198
  {
@@ -204,13 +178,13 @@
204
  "raw_files": [
205
  {
206
  "published_file": "nequip_single_0-best.ckpt",
207
- "source_file": "results/al/nequip_random_seed3/iter09_model/nequip_single_0/best.ckpt"
208
  }
209
  ],
210
- "selected_seed": 3,
211
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
212
  "source_files": [
213
- "results/al/nequip_random_seed3/iter09_model/nequip_single_0/best.ckpt"
214
  ],
215
  "strategy": "random"
216
  },
@@ -242,15 +216,26 @@
242
  ],
243
  "final_iteration": 9,
244
  "final_metrics": {
245
- "energy_mae": 0.0014821863166644108,
246
- "energy_rmse": 0.0020343181249976032,
247
- "forces_mae": 0.007570806259259011,
248
- "forces_rmse": 0.011237556111893135,
 
 
 
 
 
 
 
249
  "iteration": 9,
 
250
  "max_disagreement": 0.0,
251
  "mean_disagreement": 0.0,
252
  "n_labeled": 550,
253
- "training_time": 7280.582736492157
 
 
 
254
  },
255
  "packaged_files": [
256
  {
@@ -272,23 +257,23 @@
272
  "raw_files": [
273
  {
274
  "published_file": "committee_0-best.ckpt",
275
- "source_file": "results/al/nequip_qbc_seed1/iter09_committee/committee_0/best.ckpt"
276
  },
277
  {
278
  "published_file": "committee_1-best.ckpt",
279
- "source_file": "results/al/nequip_qbc_seed1/iter09_committee/committee_1/best.ckpt"
280
  },
281
  {
282
  "published_file": "committee_2-best.ckpt",
283
- "source_file": "results/al/nequip_qbc_seed1/iter09_committee/committee_2/best.ckpt"
284
  }
285
  ],
286
- "selected_seed": 1,
287
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
288
  "source_files": [
289
- "results/al/nequip_qbc_seed1/iter09_committee/committee_0/best.ckpt",
290
- "results/al/nequip_qbc_seed1/iter09_committee/committee_1/best.ckpt",
291
- "results/al/nequip_qbc_seed1/iter09_committee/committee_2/best.ckpt"
292
  ],
293
  "strategy": "qbc"
294
  }
 
4
  "bundle": "mace-random",
5
  "compiled_files": [
6
  {
7
+ "command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_random_seed1/iter09_model/models/mace_single_0_stagetwo.model --format libtorch",
8
  "format": "libtorch",
9
  "head": "Default",
10
  "published_file": "mace_single_0_stagetwo.model-lammps.pt",
 
13
  ],
14
  "final_iteration": 9,
15
  "final_metrics": {
16
+ "acquisition_rule": "random",
17
+ "acquisition_seed": null,
18
+ "candidate_n": 0,
19
+ "candidate_policy": "proposal_stream_filtered",
20
+ "candidate_seed": null,
21
+ "energy_mae": 0.0017917274611536414,
22
+ "energy_rmse": 0.002695636120741299,
23
+ "evaluation_policy": "fixed_member",
24
+ "forces_mae": 0.009449580705448683,
25
+ "forces_rmse": 0.014669491447455139,
26
+ "is_acquisition_iteration": false,
27
  "iteration": 9,
28
+ "labeled_n_after_acquisition": 550,
29
  "max_disagreement": 0.0,
30
  "mean_disagreement": 0.0,
31
  "n_labeled": 550,
32
+ "selected_n": 0,
33
+ "train_n_before_acquisition": 550,
34
+ "training_policy": "scratch",
35
+ "training_time": 1608.996175289154
36
  },
37
  "packaged_files": [],
38
  "raw_files": [
39
  {
40
  "published_file": "mace_single_0_stagetwo.model",
41
+ "source_file": "results/al_v4/v4_primary_fixed_member/mace_random_seed1/iter09_model/models/mace_single_0_stagetwo.model"
42
  }
43
  ],
44
+ "selected_seed": 1,
45
+ "source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
46
  "source_files": [
47
+ "results/al_v4/v4_primary_fixed_member/mace_random_seed1/iter09_model/models/mace_single_0_stagetwo.model"
48
  ],
49
  "strategy": "random"
50
  },
 
53
  "bundle": "mace-qbc",
54
  "compiled_files": [
55
  {
56
+ "command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model --format libtorch",
57
  "format": "libtorch",
58
  "head": "Default",
59
  "published_file": "mace_qbc_0_stagetwo.model-lammps.pt",
60
  "raw_file": "mace_qbc_0_stagetwo.model"
61
  },
62
  {
63
+ "command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model --format libtorch",
64
  "format": "libtorch",
65
  "head": "Default",
66
  "published_file": "mace_qbc_1_stagetwo.model-lammps.pt",
67
  "raw_file": "mace_qbc_1_stagetwo.model"
68
  },
69
  {
70
+ "command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model --format libtorch",
71
  "format": "libtorch",
72
  "head": "Default",
73
  "published_file": "mace_qbc_2_stagetwo.model-lammps.pt",
74
  "raw_file": "mace_qbc_2_stagetwo.model"
75
  },
76
  {
77
+ "command": "pixi run python -m mace.cli.create_lammps_model results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model --format libtorch",
78
  "format": "libtorch",
79
  "head": "Default",
80
  "published_file": "mace_qbc_3_stagetwo.model-lammps.pt",
 
83
  ],
84
  "final_iteration": 9,
85
  "final_metrics": {
86
+ "acquisition_rule": "uncertainty",
87
+ "acquisition_seed": null,
88
+ "candidate_n": 0,
89
+ "candidate_policy": "proposal_stream_filtered",
90
+ "candidate_seed": null,
91
+ "energy_mae": 0.001692634182771144,
92
+ "energy_rmse": 0.0022133917348856097,
93
+ "evaluation_policy": "fixed_member",
94
+ "forces_mae": 0.008605168772635183,
95
+ "forces_rmse": 0.01183799714274852,
96
+ "is_acquisition_iteration": false,
97
  "iteration": 9,
98
+ "labeled_n_after_acquisition": 550,
99
  "max_disagreement": 0.0,
100
  "mean_disagreement": 0.0,
101
  "n_labeled": 550,
102
+ "selected_n": 0,
103
+ "train_n_before_acquisition": 550,
104
+ "training_policy": "scratch",
105
+ "training_time": 6118.126273155212
106
  },
107
  "packaged_files": [],
108
  "raw_files": [
109
  {
110
  "published_file": "mace_qbc_0_stagetwo.model",
111
+ "source_file": "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model"
112
  },
113
  {
114
  "published_file": "mace_qbc_1_stagetwo.model",
115
+ "source_file": "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model"
116
  },
117
  {
118
  "published_file": "mace_qbc_2_stagetwo.model",
119
+ "source_file": "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model"
120
  },
121
  {
122
  "published_file": "mace_qbc_3_stagetwo.model",
123
+ "source_file": "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
124
  }
125
  ],
126
+ "selected_seed": 3,
127
+ "source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
128
  "source_files": [
129
+ "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_0/models/mace_qbc_0_stagetwo.model",
130
+ "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_1/models/mace_qbc_1_stagetwo.model",
131
+ "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_2/models/mace_qbc_2_stagetwo.model",
132
+ "results/al_v4/v4_primary_fixed_member/mace_qbc_seed3/iter09_committee/member_3/models/mace_qbc_3_stagetwo.model"
133
  ],
134
  "strategy": "qbc"
135
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
136
  {
137
  "architecture": "NequIP",
138
  "bundle": "nequip-random",
 
147
  ],
148
  "final_iteration": 9,
149
  "final_metrics": {
150
+ "acquisition_rule": "random",
151
+ "acquisition_seed": null,
152
+ "candidate_n": 0,
153
+ "candidate_policy": "proposal_stream_filtered",
154
+ "candidate_seed": null,
155
+ "energy_mae": 0.004235895300860648,
156
+ "energy_rmse": 0.0050607236903155136,
157
+ "evaluation_policy": "fixed_member",
158
+ "forces_mae": 0.01065873935388155,
159
+ "forces_rmse": 0.017322629771456894,
160
+ "is_acquisition_iteration": false,
161
  "iteration": 9,
162
+ "labeled_n_after_acquisition": 550,
163
  "max_disagreement": 0.0,
164
  "mean_disagreement": 0.0,
165
  "n_labeled": 550,
166
+ "selected_n": 0,
167
+ "train_n_before_acquisition": 550,
168
+ "training_policy": "scratch",
169
+ "training_time": 4275.6661376953125
170
  },
171
  "packaged_files": [
172
  {
 
178
  "raw_files": [
179
  {
180
  "published_file": "nequip_single_0-best.ckpt",
181
+ "source_file": "results/al_v4/v4_primary_fixed_member/nequip_random_seed2/iter09_model/nequip_single_0/best.ckpt"
182
  }
183
  ],
184
+ "selected_seed": 2,
185
+ "source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
186
  "source_files": [
187
+ "results/al_v4/v4_primary_fixed_member/nequip_random_seed2/iter09_model/nequip_single_0/best.ckpt"
188
  ],
189
  "strategy": "random"
190
  },
 
216
  ],
217
  "final_iteration": 9,
218
  "final_metrics": {
219
+ "acquisition_rule": "uncertainty",
220
+ "acquisition_seed": null,
221
+ "candidate_n": 0,
222
+ "candidate_policy": "proposal_stream_filtered",
223
+ "candidate_seed": null,
224
+ "energy_mae": 0.006517067205983039,
225
+ "energy_rmse": 0.007017354304601934,
226
+ "evaluation_policy": "fixed_member",
227
+ "forces_mae": 0.009533511264421878,
228
+ "forces_rmse": 0.013129118546651844,
229
+ "is_acquisition_iteration": false,
230
  "iteration": 9,
231
+ "labeled_n_after_acquisition": 550,
232
  "max_disagreement": 0.0,
233
  "mean_disagreement": 0.0,
234
  "n_labeled": 550,
235
+ "selected_n": 0,
236
+ "train_n_before_acquisition": 550,
237
+ "training_policy": "scratch",
238
+ "training_time": 17430.808257341385
239
  },
240
  "packaged_files": [
241
  {
 
257
  "raw_files": [
258
  {
259
  "published_file": "committee_0-best.ckpt",
260
+ "source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_0/best.ckpt"
261
  },
262
  {
263
  "published_file": "committee_1-best.ckpt",
264
+ "source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_1/best.ckpt"
265
  },
266
  {
267
  "published_file": "committee_2-best.ckpt",
268
+ "source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_2/best.ckpt"
269
  }
270
  ],
271
+ "selected_seed": 3,
272
+ "source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
273
  "source_files": [
274
+ "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_0/best.ckpt",
275
+ "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_1/best.ckpt",
276
+ "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_2/best.ckpt"
277
  ],
278
  "strategy": "qbc"
279
  }
nequip-qbc/metadata.json CHANGED
@@ -26,15 +26,26 @@
26
  ],
27
  "final_iteration": 9,
28
  "final_metrics": {
29
- "energy_mae": 0.0014821863166644108,
30
- "energy_rmse": 0.0020343181249976032,
31
- "forces_mae": 0.007570806259259011,
32
- "forces_rmse": 0.011237556111893135,
 
 
 
 
 
 
 
33
  "iteration": 9,
 
34
  "max_disagreement": 0.0,
35
  "mean_disagreement": 0.0,
36
  "n_labeled": 550,
37
- "training_time": 7280.582736492157
 
 
 
38
  },
39
  "packaged_files": [
40
  {
@@ -53,26 +64,31 @@
53
  "raw_file": "committee_2-best.ckpt"
54
  }
55
  ],
 
 
 
 
 
56
  "raw_files": [
57
  {
58
  "published_file": "committee_0-best.ckpt",
59
- "source_file": "results/al/nequip_qbc_seed1/iter09_committee/committee_0/best.ckpt"
60
  },
61
  {
62
  "published_file": "committee_1-best.ckpt",
63
- "source_file": "results/al/nequip_qbc_seed1/iter09_committee/committee_1/best.ckpt"
64
  },
65
  {
66
  "published_file": "committee_2-best.ckpt",
67
- "source_file": "results/al/nequip_qbc_seed1/iter09_committee/committee_2/best.ckpt"
68
  }
69
  ],
70
- "selected_seed": 1,
71
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
72
  "source_files": [
73
- "results/al/nequip_qbc_seed1/iter09_committee/committee_0/best.ckpt",
74
- "results/al/nequip_qbc_seed1/iter09_committee/committee_1/best.ckpt",
75
- "results/al/nequip_qbc_seed1/iter09_committee/committee_2/best.ckpt"
76
  ],
77
  "strategy": "qbc"
78
  }
 
26
  ],
27
  "final_iteration": 9,
28
  "final_metrics": {
29
+ "acquisition_rule": "uncertainty",
30
+ "acquisition_seed": null,
31
+ "candidate_n": 0,
32
+ "candidate_policy": "proposal_stream_filtered",
33
+ "candidate_seed": null,
34
+ "energy_mae": 0.006517067205983039,
35
+ "energy_rmse": 0.007017354304601934,
36
+ "evaluation_policy": "fixed_member",
37
+ "forces_mae": 0.009533511264421878,
38
+ "forces_rmse": 0.013129118546651844,
39
+ "is_acquisition_iteration": false,
40
  "iteration": 9,
41
+ "labeled_n_after_acquisition": 550,
42
  "max_disagreement": 0.0,
43
  "mean_disagreement": 0.0,
44
  "n_labeled": 550,
45
+ "selected_n": 0,
46
+ "train_n_before_acquisition": 550,
47
+ "training_policy": "scratch",
48
+ "training_time": 17430.808257341385
49
  },
50
  "packaged_files": [
51
  {
 
64
  "raw_file": "committee_2-best.ckpt"
65
  }
66
  ],
67
+ "published_files": [
68
+ "committee_0-best.ckpt",
69
+ "committee_1-best.ckpt",
70
+ "committee_2-best.ckpt"
71
+ ],
72
  "raw_files": [
73
  {
74
  "published_file": "committee_0-best.ckpt",
75
+ "source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_0/best.ckpt"
76
  },
77
  {
78
  "published_file": "committee_1-best.ckpt",
79
+ "source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_1/best.ckpt"
80
  },
81
  {
82
  "published_file": "committee_2-best.ckpt",
83
+ "source_file": "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_2/best.ckpt"
84
  }
85
  ],
86
+ "selected_seed": 3,
87
+ "source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
88
  "source_files": [
89
+ "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_0/best.ckpt",
90
+ "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_1/best.ckpt",
91
+ "results/al_v4/v4_primary_fixed_member/nequip_qbc_seed3/iter09_committee/committee_2/best.ckpt"
92
  ],
93
  "strategy": "qbc"
94
  }
nequip-random/metadata.json CHANGED
@@ -12,15 +12,26 @@
12
  ],
13
  "final_iteration": 9,
14
  "final_metrics": {
15
- "energy_mae": 0.004738401647407954,
16
- "energy_rmse": 0.005157445425014896,
17
- "forces_mae": 0.008885047953261107,
18
- "forces_rmse": 0.013798360721764913,
 
 
 
 
 
 
 
19
  "iteration": 9,
 
20
  "max_disagreement": 0.0,
21
  "mean_disagreement": 0.0,
22
  "n_labeled": 550,
23
- "training_time": 2944.7004199028015
 
 
 
24
  },
25
  "packaged_files": [
26
  {
@@ -29,16 +40,19 @@
29
  "raw_file": "nequip_single_0-best.ckpt"
30
  }
31
  ],
 
 
 
32
  "raw_files": [
33
  {
34
  "published_file": "nequip_single_0-best.ckpt",
35
- "source_file": "results/al/nequip_random_seed3/iter09_model/nequip_single_0/best.ckpt"
36
  }
37
  ],
38
- "selected_seed": 3,
39
- "source_commit": "8fea4f53dcdf581ca23a820ab1cc8de9708f7aac",
40
  "source_files": [
41
- "results/al/nequip_random_seed3/iter09_model/nequip_single_0/best.ckpt"
42
  ],
43
  "strategy": "random"
44
  }
 
12
  ],
13
  "final_iteration": 9,
14
  "final_metrics": {
15
+ "acquisition_rule": "random",
16
+ "acquisition_seed": null,
17
+ "candidate_n": 0,
18
+ "candidate_policy": "proposal_stream_filtered",
19
+ "candidate_seed": null,
20
+ "energy_mae": 0.004235895300860648,
21
+ "energy_rmse": 0.0050607236903155136,
22
+ "evaluation_policy": "fixed_member",
23
+ "forces_mae": 0.01065873935388155,
24
+ "forces_rmse": 0.017322629771456894,
25
+ "is_acquisition_iteration": false,
26
  "iteration": 9,
27
+ "labeled_n_after_acquisition": 550,
28
  "max_disagreement": 0.0,
29
  "mean_disagreement": 0.0,
30
  "n_labeled": 550,
31
+ "selected_n": 0,
32
+ "train_n_before_acquisition": 550,
33
+ "training_policy": "scratch",
34
+ "training_time": 4275.6661376953125
35
  },
36
  "packaged_files": [
37
  {
 
40
  "raw_file": "nequip_single_0-best.ckpt"
41
  }
42
  ],
43
+ "published_files": [
44
+ "nequip_single_0-best.ckpt"
45
+ ],
46
  "raw_files": [
47
  {
48
  "published_file": "nequip_single_0-best.ckpt",
49
+ "source_file": "results/al_v4/v4_primary_fixed_member/nequip_random_seed2/iter09_model/nequip_single_0/best.ckpt"
50
  }
51
  ],
52
+ "selected_seed": 2,
53
+ "source_commit": "9ecd37d3e1d75aa3873ea80f7202e3e54fff15e2",
54
  "source_files": [
55
+ "results/al_v4/v4_primary_fixed_member/nequip_random_seed2/iter09_model/nequip_single_0/best.ckpt"
56
  ],
57
  "strategy": "random"
58
  }