Spaces:
Sleeping
Sleeping
Download ml_model/loader.py from rriviere/attrition-api-dev: direct link, hf CLI and curl.
- Browser
- Download file 271 Bytes
-
https://huggingface.co/spaces/rriviere/attrition-api-dev/resolve/main/ml_model/loader.py
- Command line
-
hf download hf://spaces/rriviere/attrition-api-dev/ml_model/loader.py
-
curl -L -o loader.py https://huggingface.co/spaces/rriviere/attrition-api-dev/resolve/main/ml_model/loader.py
271 Bytes
| 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"] | |