0.90.4

felixbur · Oct 15, 2024 · 56ea22b · 56ea22b
1 parent 533a317
commit 56ea22b
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,10 @@
 Changelog
 =========
 
+Version 0.90.4
+--------------
+* added plot format for multidb
+
 Version 0.90.3
 --------------
 * refactorings and documentations

diff --git a/nkululeko/constants.py b/nkululeko/constants.py
@@ -1,2 +1,2 @@
-VERSION="0.90.3"
+VERSION="0.90.4"
 SAMPLING_RATE = 16000
diff --git a/nkululeko/multidb.py b/nkululeko/multidb.py
@@ -115,7 +115,11 @@ def main(src_dir):
     print(repr(results))
     print(repr(last_epochs))
     root = os.path.join(config["EXP"]["root"], "")
-    plot_name = f"{root}/heatmap.png"
+    try:
+        format = config["PLOT"]["format"]
+        plot_name = f"{root}/heatmap.{format}"
+    except KeyError:
+        plot_name = f"{root}/heatmap.png"
     plot_heatmap(results, last_epochs, datasets, plot_name, config, datasets)