Syntaxe d'évaluation d'un modèle de régression avec scikit-learn
bonjour,
je suis sur un sujet de machinie learning et je dois tester mon modèle avec les métriques de régression linéaire. MSE, R2 et MAE je n'arrive pas à trouver la bonne syntaxe en utilisant scikit learn comme bibliothèque. je ne sais pas si vous pouvez m'aider à trouver la bonne syntaxe pour l'évaluation de mon modèle.
Merci
Salut,
Je n'ai jamais utilisé scikit, mais sur internet il y pas mal d'exemples :
from sklearn.metrics import mean_squared_error, r2_score, mean_absolute_error
# y_true : les vraies valeurs
# y_pred : les prédictions de votre modèle
# Calcul des métriques
mse = mean_squared_error(y_true, y_pred)
r2 = r2_score(y_true, y_pred)
mae = mean_absolute_error(y_true, y_pred)
print(f"MSE (Mean Squared Error): {mse}")
print(f"R² (Coefficient de détermination): {r2}")
print(f"MAE (Mean Absolute Error): {mae}")
Sur des sites anglophones tu as quelques exemples.
Tu peux peut-être demander sur le discord.
J'espère que ça peut t'aider ^^
Inscris-toi
(c'est gratuit !)
Tu dois créer un compte pour participer aux discussions.
Créer un compte