Pret-a-depenser / scripts /export_model.py
celpri
Update monitoring dashboard
29fec18
Raw
History Blame Contribute Delete
508 Bytes
import os
import mlflow
import mlflow.sklearn
import joblib
BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
DB_PATH = os.path.join(BASE_DIR, "mlflow.db")
mlflow.set_tracking_uri(f"sqlite:///{DB_PATH}")
MODEL_URI = "models:/CreditScoring_LightGBM/Production"
OUT_PATH = os.path.join(BASE_DIR, "app", "model.joblib")
os.makedirs(os.path.dirname(OUT_PATH), exist_ok=True)
model = mlflow.sklearn.load_model(MODEL_URI)
joblib.dump(model, OUT_PATH)
print("Export OK ->", OUT_PATH)