summaryrefslogtreecommitdiff
path: root/bin
diff options
context:
space:
mode:
authorBirte Kristina Friesel <birte.friesel@uos.de>2023-12-22 07:58:04 +0100
committerBirte Kristina Friesel <birte.friesel@uos.de>2023-12-22 07:58:04 +0100
commite5afb3394e7034482927e4155e3f9eddbbf78851 (patch)
treeedb6e12116dd72408aabbc091a847edf132177e6 /bin
parent5fd55e1a09b5558064c6eac77b2c5b39624a0739 (diff)
analyze-log: add --show-model-size support
Diffstat (limited to 'bin')
-rwxr-xr-xbin/analyze-log.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/bin/analyze-log.py b/bin/analyze-log.py
index 4c37928..2192a2e 100755
--- a/bin/analyze-log.py
+++ b/bin/analyze-log.py
@@ -267,6 +267,9 @@ def main():
error_metric=args.error_metric,
)
+ if args.show_model_size:
+ dfatool.cli.print_model_size(model)
+
if args.export_model:
print(f"Exportding model to {args.export_model}")
json_model = model.to_json()