diff options
| -rwxr-xr-x | bin/analyze-archive.py | 3 | ||||
| -rwxr-xr-x | bin/analyze-kconfig.py | 3 | ||||
| -rw-r--r-- | lib/cli.py | 24 |
3 files changed, 30 insertions, 0 deletions
diff --git a/bin/analyze-archive.py b/bin/analyze-archive.py index 5cc01f6..88b6802 100755 --- a/bin/analyze-archive.py +++ b/bin/analyze-archive.py @@ -1048,6 +1048,9 @@ if __name__ == "__main__": ] ) + if args.show_model_size: + dfatool.cli.print_model_size(model) + if args.plot_param: for kv in args.plot_param.split(";"): try: diff --git a/bin/analyze-kconfig.py b/bin/analyze-kconfig.py index 8e60472..b9f071f 100755 --- a/bin/analyze-kconfig.py +++ b/bin/analyze-kconfig.py @@ -307,6 +307,9 @@ def main(): param_values = model.distinct_param_values_by_name[name][i] print(f" Parameter {param} ∈ {param_values}") + if args.show_model_size: + dfatool.cli.print_model_size(model) + if args.export_model: with open("nfpkeys.json", "r") as f: nfpkeys = json.load(f) @@ -51,6 +51,25 @@ def print_splitinfo(param_names, info, prefix=""): print(f"{prefix}: UNKNOWN") +def print_model_size(model): + for name in model.names: + for attribute in model.attributes(name): + try: + num_nodes = model.attr_by_name[name][ + attribute + ].model_function.get_number_of_nodes() + max_depth = model.attr_by_name[name][ + attribute + ].model_function.get_max_depth() + print( + f"{name:15s} {attribute:20s}: {num_nodes:6d} nodes @ {max_depth:3d} max depth" + ) + except AttributeError: + print( + f"{name:15s} {attribute:20s}: {model.attr_by_name[name][attribute].model_function}" + ) + + def format_quality_measures(result): if "smape" in result: return "{:6.2f}% / {:9.0f}".format(result["smape"], result["mae"]) @@ -104,6 +123,11 @@ def add_standard_arguments(parser): help="Export model and model quality to LaTeX dataref file", ) parser.add_argument( + "--show-model-size", + action="store_true", + help="Show model size (e.g. regression tree height and node count)", + ) + parser.add_argument( "--cross-validate", metavar="<method>:<count>", type=str, |