Skip to content

Commit

Permalink
Update benchmark results.
Browse files Browse the repository at this point in the history
  • Loading branch information
PatReis committed Jan 5, 2024
1 parent 9025882 commit a6f495f
Showing 1 changed file with 22 additions and 9 deletions.
31 changes: 22 additions & 9 deletions training/results/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -158,6 +158,7 @@ Materials Project dataset from Matbench with 4764 crystal structures and their c
| DimeNetPP.make_crystal_model | 4.0.0 | 780 | 0.3415 ± 0.0542 | 1.9637 ± 0.6323 |
| Megnet.make_crystal_model | 4.0.0 | 1000 | 0.3362 ± 0.0550 | 2.0156 ± 0.5872 |
| NMPN.make_crystal_model | 4.0.0 | 700 | 0.3289 ± 0.0489 | 1.8770 ± 0.6522 |
| PAiNN.make_crystal_model | 4.0.0 | 800 | 0.3539 ± 0.0433 | 1.8661 ± 0.5984 |
| Schnet.make_crystal_model | 4.0.0 | 800 | **0.3180 ± 0.0359** | **1.8509 ± 0.5854** |

#### MatProjectEFormDataset
Expand Down Expand Up @@ -203,6 +204,7 @@ Materials Project dataset from Matbench with 10987 crystal structures and their

| model | kgcnn | epochs | MAE [log(GPa)] | RMSE [log(GPa)] |
|:--------------------------|:--------|---------:|:-----------------------|:-----------------------|
| CGCNN.make_crystal_model | 4.0.0 | 1000 | 0.0874 ± 0.0022 | 0.1354 ± 0.0056 |
| Schnet.make_crystal_model | 4.0.0 | 800 | **0.0836 ± 0.0021** | **0.1296 ± 0.0044** |

#### MatProjectLogKVRHDataset
Expand All @@ -217,17 +219,27 @@ Materials Project dataset from Matbench with 10987 crystal structures and their

Materials Project dataset from Matbench with 18928 crystal structures and their corresponding Heat of formation of the entire 5-atom perovskite cell in eV. We use a random 5-fold cross-validation.

| model | kgcnn | epochs | MAE [eV] | RMSE [eV] |
|:--------------------------|:--------|---------:|:-----------------------|:-----------------------|
| Schnet.make_crystal_model | 4.0.0 | 800 | **0.0381 ± 0.0005** | **0.0645 ± 0.0024** |
| model | kgcnn | epochs | MAE [eV] | RMSE [eV] |
|:-----------------------------|:--------|---------:|:-----------------------|:-----------------------|
| CGCNN.make_crystal_model | 4.0.0 | 1000 | 0.0425 ± 0.0011 | 0.0712 ± 0.0037 |
| DimeNetPP.make_crystal_model | 4.0.0 | 780 | 0.0447 ± 0.0016 | 0.0730 ± 0.0050 |
| Megnet.make_crystal_model | 4.0.0 | 1000 | 0.0388 ± 0.0017 | 0.0675 ± 0.0041 |
| NMPN.make_crystal_model | 4.0.0 | 700 | **0.0381 ± 0.0009** | 0.0652 ± 0.0029 |
| PAiNN.make_crystal_model | 4.0.0 | 800 | 0.0474 ± 0.0003 | 0.0762 ± 0.0017 |
| Schnet.make_crystal_model | 4.0.0 | 800 | 0.0381 ± 0.0005 | **0.0645 ± 0.0024** |

#### MatProjectPhononsDataset

Materials Project dataset from Matbench with 1,265 crystal structures and their corresponding vibration properties in [1/cm]. We use a random 5-fold cross-validation.

| model | kgcnn | epochs | MAE [eV/atom] | RMSE [eV/atom] |
|:--------------------------|:--------|---------:|:------------------------|:-------------------------|
| Schnet.make_crystal_model | 4.0.0 | 800 | **43.0692 ± 3.6227** | **88.5151 ± 20.0244** |
| model | kgcnn | epochs | MAE [eV/atom] | RMSE [eV/atom] |
|:-----------------------------|:--------|---------:|:------------------------|:-------------------------|
| CGCNN.make_crystal_model | 4.0.0 | 1000 | 42.6447 ± 4.5721 | 92.1627 ± 21.4345 |
| DimeNetPP.make_crystal_model | 4.0.0 | 780 | 39.8893 ± 3.1280 | 77.5776 ± 16.0908 |
| Megnet.make_crystal_model | 4.0.0 | 1000 | **30.6620 ± 2.9013** | **60.8733 ± 17.1448** |
| NMPN.make_crystal_model | 4.0.0 | 700 | 45.9344 ± 5.7908 | 95.4136 ± 35.5401 |
| PAiNN.make_crystal_model | 4.0.0 | 800 | 47.5408 ± 4.2815 | 86.6761 ± 11.9220 |
| Schnet.make_crystal_model | 4.0.0 | 800 | 43.0692 ± 3.6227 | 88.5151 ± 20.0244 |

#### MUTAGDataset

Expand Down Expand Up @@ -286,9 +298,10 @@ QM7 dataset is a subset of GDB-13. Molecules of up to 23 atoms (including 7 heav

QM9 dataset of 134k stable small organic molecules made up of C,H,O,N,F. Labels include geometric, energetic, electronic, and thermodynamic properties. We use a random 5-fold cross-validation, but not all splits are evaluated for cheaper evaluation. Test errors are MAE and for energies are given in [eV].

| model | kgcnn | epochs | HOMO [eV] | LUMO [eV] | U0 [eV] | H [eV] | G [eV] |
|:--------|:--------|---------:|:-----------------------|:-----------------------|:-----------------------|:-----------------------|:-----------------------|
| Schnet | 4.0.0 | 800 | **0.0402 ± 0.0007** | **0.0340 ± 0.0001** | **0.0142 ± 0.0002** | **0.0146 ± 0.0002** | **0.0143 ± 0.0002** |
| model | kgcnn | epochs | HOMO [eV] | LUMO [eV] | U0 [eV] | H [eV] | G [eV] |
|:--------|:--------|---------:|:-----------------------|:-----------------------|:-----------------------|:-----------------------|:-------------------|
| PAiNN | 4.0.0 | 872 | 0.0483 ± 0.0275 | **0.0268 ± 0.0002** | **0.0099 ± 0.0003** | **0.0101 ± 0.0003** | **nan ± nan** |
| Schnet | 4.0.0 | 800 | **0.0402 ± 0.0007** | 0.0340 ± 0.0001 | 0.0142 ± 0.0002 | 0.0146 ± 0.0002 | 0.0143 ± 0.0002 |

#### SIDERDataset

Expand Down

0 comments on commit a6f495f

Please sign in to comment.