From 7e07dfbe880e380b95586ce9eb32bf80f14736c1 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Wed, 24 Nov 2021 15:16:26 +0100 Subject: validation: add SMAPE output --- lib/validation.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lib/validation.py b/lib/validation.py index 81b1819..47395dc 100644 --- a/lib/validation.py +++ b/lib/validation.py @@ -218,6 +218,7 @@ class CrossValidator: ret[name][attribute] = { "mae_list": list(), "rmsd_list": list(), + "mape_list": list(), "smape_list": list(), } @@ -225,14 +226,14 @@ class CrossValidator: res = self._single_xv(model_getter, training_and_validation_by_name) for name in self.names: for attribute in self.by_name[name]["attributes"]: - for measure in ("mae", "rmsd", "smape"): + for measure in ("mae", "rmsd", "mape", "smape"): ret[name][attribute][f"{measure}_list"].append( res[name][attribute][measure] ) for name in self.names: for attribute in self.by_name[name]["attributes"]: - for measure in ("mae", "rmsd", "smape"): + for measure in ("mae", "rmsd", "mape", "smape"): ret[name][attribute][measure] = np.mean( ret[name][attribute][f"{measure}_list"] ) -- cgit v1.2.3