From 7ee09399e7cba10f693ec301961637a9f736750b Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Wed, 20 Dec 2023 13:45:29 +0100 Subject: model quality table: roughly indicate well-fitting / badly fitting models --- lib/cli.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'lib/cli.py') diff --git a/lib/cli.py b/lib/cli.py index 78fc15c..041e524 100644 --- a/lib/cli.py +++ b/lib/cli.py @@ -176,6 +176,20 @@ def model_quality_table(lut, model, static, model_info, xv_method=None): buf += format_quality_measures(result) else: buf += f"""{"----":>7s} """ + if type(model_info(key, attr)) is not StaticFunction: + if model[key][attr]["mae"] > static[key][attr]["mae"]: + buf += " :-(" + elif ( + model[key][attr]["mae"] <= 2 * lut[key][attr]["mae"] + and static[key][attr]["mae"] > 4 * lut[key][attr]["mae"] + ): + buf += " :-D" + elif ( + static[key][attr]["mae"] - model[key][attr]["mae"] + > model[key][attr]["mae"] - lut[key][attr]["mae"] + and static[key][attr]["mae"] > 1.1 * lut[key][attr]["mae"] + ): + buf += " :-)" print(buf) -- cgit v1.2.3