From 344cff131aee49594f3cf2f61ad729f7a00a6756 Mon Sep 17 00:00:00 2001 From: Anna Foix Date: Mon, 22 Jul 2024 20:12:45 +0100 Subject: [PATCH] fix model name in shapeembed output csv --- scripts/shapeembed/shapeembed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/shapeembed/shapeembed.py b/scripts/shapeembed/shapeembed.py index 744f29fa..b027a3c9 100755 --- a/scripts/shapeembed/shapeembed.py +++ b/scripts/shapeembed/shapeembed.py @@ -104,7 +104,7 @@ def tag_cols(params): cols = [] cols.append(('dataset', params.dataset.name)) - cols.append(('model', model_str(params))) + cols.append(('model', params.model_name)) for k, v in vars(params.model_args).items(): cols.append((k, v)) cols.append(('compression_factor', params.compression_factor)) cols.append(('latent_dim', params.latent_dim))