From 59e35e5112db018a3ba150c4a661292a154398e3 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Thu, 11 Jan 2024 14:07:56 +0100 Subject: move --show-quality=table header to cli.py --- lib/cli.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'lib') diff --git a/lib/cli.py b/lib/cli.py index 3fa39b3..25ad260 100644 --- a/lib/cli.py +++ b/lib/cli.py @@ -181,7 +181,7 @@ def format_quality_measures(result, error_metric="smape", col_len=8): def model_quality_table( - lut, model, static, model_info, xv_method=None, error_metric="smape" + lut, model, static, model_info, xv_method=None, xv_count=None, error_metric="smape" ): key_len = len("Key") attr_len = len("Attribute") @@ -201,6 +201,13 @@ def model_quality_table( else: xv_header = "training" + if xv_method is not None: + print( + f"Model error ({error_metric}) after cross validation ({xv_method}, {xv_count}):" + ) + else: + print(f"Model error ({error_metric}) on training data:") + print( f"""{"":>{key_len}s} {"":>{attr_len}s} {"training":>8s} {xv_header:>8s} {xv_header:>8s}""" ) -- cgit v1.2.3