ACA050's picture
Update backend/utils/model_io.py
044c9a2 verified
raw
history blame contribute delete
224 Bytes
import os
import joblib
class ModelIO:
def save(self, model, path):
os.makedirs(os.path.dirname(path), exist_ok=True)
joblib.dump(model, path)
def load(self, path):
return joblib.load(path)