Published 2026 | Version v1.0
Model Open

Goldilocks CGCNN metallicity classifier model(Matbench mp_is_metal)

  • 1. UKRI STFC
  • 1. UKRI STFC

Description

A crystal graph convolutional neural network that predicts whether a periodic material has a zero DFT band gap. Goldilocks uses metallicity to decide how densely reciprocal space must be sampled and whether smearing applies.

Trained on the Matbench mp_is_metal task: 106113 Materials Project structures, split 70/10/10/10 by reduced composition so that polymorphs of one composition cannot straddle the split, and stratified by label. Training stops on validation ROC-AUC rather than loss. On the 10625-structure test split it reaches 0.748 accuracy, 0.569 Matthews correlation and 0.951 ROC-AUC, against 0.544 accuracy and 0.000 MCC for a majority-class baseline.

The decision threshold is 0.0478, not 0.5. It is chosen on the validation split subject to a floor of 0.97 recall, because missing a metal understates the mesh a calculation needs and produces a wrong answer that does not look wrong, while an unnecessary dense mesh only costs compute. Test recall is 0.972 and test precision 0.649: roughly a third of what this model calls metallic is not, and that is a deliberate trade.

The record contains the weights and model.json, which records the architecture, the feature contract, the target contract, the threshold and the rule that chose it, the digests, and the atomic embedding table it requires from record m742g-g0k14. It was written by the run that fitted the model.

Files

README.md

Files (553.9 kB)

Name Size Download all
md5:29a97daf8b25c28d137a8ebff6c2f812
28.4 kB Preview Download
md5:f76a56d0d6aba9a76b0e7f5393a4441c
502.0 kB Download
md5:3a418addfcee294b3ab838d40539e4eb
851 Bytes Preview Download
md5:1e2d47fa56fdedd32d26e7ebf89bbd3c
16.7 kB Preview Download
md5:e84751973041ab67628c5a423c61f0f6
6.0 kB Preview Download