Upload 3 files
Browse files- README.md +19 -0
- inference.py +14 -0
- model.joblib +3 -0
README.md
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Mon Modèle Scikit-learn
|
2 |
+
Ce modèle a été entraîné avec scikit-learn pour prédire les émissions en CO2 d'une voiture.
|
3 |
+
|
4 |
+
L'entrainement du modèle a été fait avec Google Collab : https://colab.research.google.com/drive/1n1Vtx6873e06c0WYG9JKIJ_ZLTMkptcX#scrollTo=_eLI1udbyDT3
|
5 |
+
|
6 |
+
La documentation sur l'entraînement du modèle est la suivante : https://docs.google.com/spreadsheets/d/1oBshoNy2NJZQreOEbBfcWCCg2wo0PQDo/edit?gid=1220399676#gid=1220399676
|
7 |
+
|
8 |
+
## Exemple d'utilisation
|
9 |
+
```python
|
10 |
+
import joblib
|
11 |
+
import numpy as np
|
12 |
+
|
13 |
+
# Charger le modèle
|
14 |
+
model = joblib.load("model.joblib")
|
15 |
+
|
16 |
+
# Effectuer une prédiction
|
17 |
+
input_data = [5.1, 3.5, 1.4, 0.2] # Exemple pour l'iris
|
18 |
+
prediction = model.predict([input_data])
|
19 |
+
print(f"Prédiction : {prediction}")
|
inference.py
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import joblib
|
2 |
+
import numpy as np
|
3 |
+
|
4 |
+
# Charger le modèle
|
5 |
+
model = joblib.load("model.joblib")
|
6 |
+
|
7 |
+
# Fonction de prédiction
|
8 |
+
def predict(input_data):
|
9 |
+
"""
|
10 |
+
input_data : array-like ou liste de caractéristiques
|
11 |
+
Retourne les prédictions du modèle
|
12 |
+
"""
|
13 |
+
input_array = np.array(input_data).reshape(1, -1)
|
14 |
+
return model.predict(input_array).tolist()
|
model.joblib
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:67758c44ac783c45468857aec2cc3fb1b58309d2aa0b2a5cf211f91053c4d73d
|
3 |
+
size 236318305
|