create output dir on run

mayu-ot · Aug 20, 2020 · 587e73c · 587e73c
1 parent 0b1c5f8
commit 587e73c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/experiments/blind_baselines.py b/src/experiments/blind_baselines.py
@@ -1,5 +1,6 @@
 from typing import Dict
 import pandas as pd
+import os
 import json
 import numpy as np
 import matplotlib.pyplot as plt
@@ -107,6 +108,10 @@ def plot_performance_summary(
     plt.legend()
     plt.title(title)
     sns.despine(left=False)
+
+    if not os.path.exists(os.path.dirname(outfile)):
+        os.makedirs(os.path.dirname(outfile))
+
     plt.savefig(outfile, bbox_inches="tight")