From 5b9a8b02ab7635406a86ffd58f0943b1c8f441b3 Mon Sep 17 00:00:00 2001 From: Davide Tisi Date: Wed, 20 Nov 2024 15:22:04 +0100 Subject: [PATCH] added a Legend of the definitions showed in the logger (#395) * added a Legend of the definitions showed in the logger --- docs/src/architectures/soap-bpnn.rst | 2 +- src/metatrain/experimental/pet/trainer.py | 16 +++++++++++++++- 2 files changed, 16 insertions(+), 2 deletions(-) diff --git a/docs/src/architectures/soap-bpnn.rst b/docs/src/architectures/soap-bpnn.rst index f8e1ab607..21aea43b4 100644 --- a/docs/src/architectures/soap-bpnn.rst +++ b/docs/src/architectures/soap-bpnn.rst @@ -9,7 +9,7 @@ SOAP-BPNN This is a Behler-Parrinello neural network :footcite:p:`behler_generalized_2007` with using features based on the Smooth overlab of atomic positions (SOAP) -:footcite:p:`bartok_representing_2013`. The SOAP features are calculated wit `rascaline +:footcite:p:`bartok_representing_2013`. The SOAP features are calculated with `rascaline `_. Installation diff --git a/src/metatrain/experimental/pet/trainer.py b/src/metatrain/experimental/pet/trainer.py index afe8c09dd..dbe6f1e1e 100644 --- a/src/metatrain/experimental/pet/trainer.py +++ b/src/metatrain/experimental/pet/trainer.py @@ -117,7 +117,7 @@ def train( checkpoint_path = None ######################################## - # STARTNG THE PURE PET TRAINING SCRIPT # + # STARTING THE PURE PET TRAINING SCRIPT # ######################################## logging.info("Initializing PET training...") @@ -165,6 +165,20 @@ def train( f"CUDA is deterministic: {FITTING_SCHEME.CUDA_DETERMINISTIC}" ) + st = """ +Legend: LR -> Learning Rate + MAE -> Mean Square Error + RMSE -> Root Mean Square Error + V-E-MAE/at -> MAE of the Energy per atom on the Validation set + V-E-RMSE/at -> RMSE of the Energy per atom on the Validation set + V-F-MAE -> MAE of the Forces on the Validation set + V-F-RMSE -> RMSE of the Forces on the Validation set + T-E-MAE/at -> MAE of the Energy per atom on the Training set + T-E-RMSE/at -> RMSE of the Energy per atom on the Training set + T-F-MAE -> MAE of the Forces on the Training set + T-F-RMSE -> RMSE of the Forces on the Training set +Units of the Energy and Forces are the same units given in input""" + training_configuration_log += st logging.info(training_configuration_log) set_reproducibility(