diff --git a/requirement.txt b/requirement.txt index f0fe2d5..db00aff 100644 --- 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 \ No newline at end of file +seaborn==0.13.0 +critdd==0.0.4.dev0 \ No newline at end of file diff --git a/results.py b/results.py index 49236c3..e5caeda 100644 --- 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)