diff options
| -rw-r--r-- | lib/validation.py | 5 |
1 files 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"] ) |