ACA050 commited on
Commit
044c9a2
·
verified ·
1 Parent(s): ac573e4

Update backend/utils/model_io.py

Browse files
Files changed (1) hide show
  1. backend/utils/model_io.py +10 -12
backend/utils/model_io.py CHANGED
@@ -1,12 +1,10 @@
1
- import joblib
2
-
3
- class ModelIO:
4
- def save(self, model, path):
5
- joblib.dump(model, path)
6
-
7
- def load(self, path):
8
- return joblib.load(path)
9
-
10
-
11
-
12
-
 
1
+ import os
2
+ import joblib
3
+
4
+ class ModelIO:
5
+ def save(self, model, path):
6
+ os.makedirs(os.path.dirname(path), exist_ok=True)
7
+ joblib.dump(model, path)
8
+
9
+ def load(self, path):
10
+ return joblib.load(path)