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
|