Added missing folders/import for the results to run smoothly

erusseil · Feb 14, 2024 · 4b021b0 · 4b021b0
1 parent dff0cb9
commit 4b021b0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/requirement.txt b/requirement.txt
@@ -3,4 +3,5 @@ matplotlib==3.7.3
 numpy==1.24.4
 pandas==2.0.3
 scikit-learn==1.3.2
-seaborn==0.13.0
+seaborn==0.13.0
+critdd==0.0.4.dev0
diff --git a/results.py b/results.py
@@ -4,6 +4,7 @@
 import seaborn as sns
 from sympy import sympify, Symbol, exp, log
 import matplotlib.colors
+import os
 
 plt.rcParams["font.size"] = 20
 
@@ -69,6 +70,9 @@ def treesize(expr):
         tot += treesize(a)
     return tot
 
+for createfolders in ['boxplots', 'plot', 'heat']:
+    if not os.path.exists(createfolders):
+            os.makedirs(createfolders)
 
 for d in datasets:
     print(d)