File size: 271 Bytes
e59e621
 
 
100ddbe
e59e621
 
 
e406ddb
e59e621
 
1
2
3
4
5
6
7
8
9
10
11
import joblib
from pathlib import Path

MODEL_PATH = Path("ml_model/pipeline.pkl")


def load_pipeline():
    # Retourne le pipeline sklearn et le seuil de décision depuis le fichier .pkl
    data = joblib.load(MODEL_PATH)
    return data["pipeline"], data["threshold"]