avraux commited on
Commit
a0fb68d
1 Parent(s): 6f6adf8

Upload 3 files

Browse files
Files changed (3) hide show
  1. README.md +19 -0
  2. inference.py +14 -0
  3. 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