Revisions

cli/persistence.py

@@ -0,0 +1,18 @@
+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))