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() | |