fix model name in shapeembed output csv

uhlmanngroup · Jul 22, 2024 · 344cff1 · 344cff1
1 parent cecde4e
commit 344cff1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))