From f1dfa367044344760331d79361363ba1860f2367 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Mon, 1 Jul 2024 11:02:49 +0200 Subject: mutual_information: handle skipped parameters; return a dict --- bin/analyze-log.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'bin') diff --git a/bin/analyze-log.py b/bin/analyze-log.py index 50c0344..43a7d11 100755 --- a/bin/analyze-log.py +++ b/bin/analyze-log.py @@ -286,10 +286,11 @@ def main(): for name in model.names: for attr in model.attributes(name): mutual_information = model.mutual_information(name, attr) - for i, param in enumerate(model.parameters): - dref[f"mutual information/{name}/{attr}/{param}"] = ( - mutual_information[i] - ) + for param in model.parameters: + if param in mutual_information: + dref[f"mutual information/{name}/{attr}/{param}"] = ( + mutual_information[param] + ) dfatool.cli.export_dataref( args.export_dref, dref, precision=args.dref_precision -- cgit v1.2.3