19 lines
387 B
Python
19 lines
387 B
Python
import os
|
|
|
|
import joblib
|
|
|
|
MODEL_FILE = "model.joblib"
|
|
|
|
|
|
def is_saved_model(path):
|
|
return os.path.isdir(path) and os.path.exists(os.path.join(path, MODEL_FILE))
|
|
|
|
|
|
def save_model(clf, directory):
|
|
os.makedirs(directory, exist_ok=True)
|
|
joblib.dump(clf, os.path.join(directory, MODEL_FILE))
|
|
|
|
|
|
def load_model(directory):
|
|
return joblib.load(os.path.join(directory, MODEL_FILE))
|