From 56c688ace423fc29be23183b5bacd8e2671dfa2b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=EC=9D=B4=EC=A0=95=ED=99=98=20=28Lee=20Jeonghwan=29?=
 <63226383+JH-lee95@users.noreply.github.com>
Date: Wed, 18 May 2022 12:12:03 +0900
Subject: [PATCH] creating models folder

---
 examples/test_ml_classification.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/examples/test_ml_classification.py b/examples/test_ml_classification.py
index 40bba84..72c9301 100644
--- a/examples/test_ml_classification.py
+++ b/examples/test_ml_classification.py
@@ -1,5 +1,6 @@
 from treform.document_classification.ml_textclassification import documentClassifier
 import treform as ptm
+import os
 
 if __name__ == '__main__':
     document_classifier = documentClassifier()
@@ -27,6 +28,10 @@
     model_name = ml_algorithms[7]
 
     #document category and id map
+    
+    if not os.path.exists("../models/"):
+        os.mkdir("../models/")
+    
     id_category_json = '../models/ml_id_category.json'
 
     #mode is either train or predict