Update visual.py

simplify my code according to author's sugg
YujinK-CHN · Dec 13, 2021 · b0e5da5 · b0e5da5
1 parent ff9d48d
commit b0e5da5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/visual.py b/visual.py
@@ -14,8 +14,7 @@ def show_tfidf(tfidf, vocab, filename):
     plt.tight_layout()
     # creating the output folder 
     output_folder = './visual/results/'
-    if not os.path.exists(output_folder):
-       os.makedirs(output_folder)
+    os.makedirs(output_folder, exist_ok=True)
     plt.savefig(os.path.join(output_folder, '%s.png') % filename, format="png", dpi=500)
     plt.show()