co2prediction / inference.py
avraux's picture
Upload 3 files
a0fb68d verified
raw
history blame
359 Bytes
import joblib
import numpy as np
# Charger le modèle
model = joblib.load("model.joblib")
# Fonction de prédiction
def predict(input_data):
"""
input_data : array-like ou liste de caractéristiques
Retourne les prédictions du modèle
"""
input_array = np.array(input_data).reshape(1, -1)
return model.predict(input_array).tolist()