Upload folder using huggingface_hub
Browse files- README.md +20 -0
- model/MLmodel +20 -0
- model/conda.yaml +16 -0
- model/model.pkl +3 -0
- model/python_env.yaml +7 -0
- model/requirements.txt +9 -0
README.md
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- employee-turnover
|
| 4 |
+
- xgboost
|
| 5 |
+
- mlflow
|
| 6 |
+
- classification
|
| 7 |
+
library_name: scikit-learn
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# Employee Turnover Prediction Model
|
| 11 |
+
|
| 12 |
+
Modèle XGBoost pour prédire le turnover des employés.
|
| 13 |
+
|
| 14 |
+
## Utilisation
|
| 15 |
+
|
| 16 |
+
```python
|
| 17 |
+
import mlflow
|
| 18 |
+
model = mlflow.sklearn.load_model("hf://ASI-Engineer/employee-turnover-model")
|
| 19 |
+
predictions = model.predict(X)
|
| 20 |
+
```
|
model/MLmodel
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
flavors:
|
| 2 |
+
python_function:
|
| 3 |
+
env:
|
| 4 |
+
conda: conda.yaml
|
| 5 |
+
virtualenv: python_env.yaml
|
| 6 |
+
loader_module: mlflow.sklearn
|
| 7 |
+
model_path: model.pkl
|
| 8 |
+
predict_fn: predict
|
| 9 |
+
python_version: 3.12.3
|
| 10 |
+
sklearn:
|
| 11 |
+
code: null
|
| 12 |
+
pickled_model: model.pkl
|
| 13 |
+
serialization_format: cloudpickle
|
| 14 |
+
sklearn_version: 1.6.1
|
| 15 |
+
mlflow_version: 3.8.0
|
| 16 |
+
model_id: null
|
| 17 |
+
model_size_bytes: 220543
|
| 18 |
+
model_uuid: 2d0516757a5e44b59a76f41795abfe7c
|
| 19 |
+
prompts: null
|
| 20 |
+
utc_time_created: '2025-12-25 21:40:40.507658'
|
model/conda.yaml
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
channels:
|
| 2 |
+
- conda-forge
|
| 3 |
+
dependencies:
|
| 4 |
+
- python=3.12.3
|
| 5 |
+
- pip<=25.3
|
| 6 |
+
- pip:
|
| 7 |
+
- mlflow==3.8.0
|
| 8 |
+
- cloudpickle==3.1.2
|
| 9 |
+
- imbalanced-learn==0.13.0
|
| 10 |
+
- numpy==2.4.0
|
| 11 |
+
- pandas==2.3.3
|
| 12 |
+
- pytest==9.0.1
|
| 13 |
+
- scikit-learn==1.6.1
|
| 14 |
+
- scipy==1.16.3
|
| 15 |
+
- xgboost==2.1.4
|
| 16 |
+
name: mlflow-env
|
model/model.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6e51ada1b28e1a028f547cde968f04b83d2bb18dd8613f8dfbdb7ad7a279f4c2
|
| 3 |
+
size 220543
|
model/python_env.yaml
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
python: 3.12.3
|
| 2 |
+
build_dependencies:
|
| 3 |
+
- pip==25.3
|
| 4 |
+
- setuptools
|
| 5 |
+
- wheel
|
| 6 |
+
dependencies:
|
| 7 |
+
- -r requirements.txt
|
model/requirements.txt
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
mlflow==3.8.0
|
| 2 |
+
cloudpickle==3.1.2
|
| 3 |
+
imbalanced-learn==0.13.0
|
| 4 |
+
numpy==2.4.0
|
| 5 |
+
pandas==2.3.3
|
| 6 |
+
pytest==9.0.1
|
| 7 |
+
scikit-learn==1.6.1
|
| 8 |
+
scipy==1.16.3
|
| 9 |
+
xgboost==2.1.4
|