ارزیابی رگرسیون

معیارهای ارزیابی مدل رگرسیون: MAE، MSE، RMSE و R² با مثال و تفسیر

چگونه بفهمیم مدل رگرسیون خوب است؟

در دسته‌بندی «درست/غلط» داریم، اما در رگرسیون پیش‌بینی «نزدیک به واقعیت» است. معیارهای اصلی:

۱. خطای مطلق میانگین (MAE)

میانگین قدرمطلق خطاها:

MAE = میانگین(|y_واقعی - y_پیش‌بینی|)

مزیت: واحدش همان واحد برچسب است و قابل درک: «به‌طور میانگین ۳ میلیون خطا داریم».

۲. خطای مربعی میانگین (MSE)

MSE = میانگین((y_واقعی - y_پیش‌بینی)²)

مزیت: خطاهای بزرگ را به‌شدت جریمه می‌کند. عیب: واحدش مجذور است.

۳. ریشه MSE (RMSE)

RMSE = √MSE

جریمه خطای بزرگ + واحد قابل درک. محبوب‌ترین معیار در مسابقات.

۴. ضریب تعیین (R²)

نسبت واریانس توضیح‌داده‌شده توسط مدل:

مقایسه MAE و RMSE

اگر RMSE خیلی بزرگ‌تر از MAE باشد، مدل روی برخی نمونه‌ها خطای خیلی بزرگ دارد (احتمالاً پرت‌ها).

پیاده‌سازی

from sklearn.metrics import mean_absolute_error, mean_squared_error, r2_score

mae = mean_absolute_error(y_test, y_pred)
mse = mean_squared_error(y_test, y_pred)
rmse = mean_squared_error(y_test, y_pred) ** 0.5
r2 = r2_score(y_test, y_pred)

جدول مقایسه

معیارمزیتعیب
MAEقابل درک، مقاوم به پرتخطای بزرگ را کم‌اهمیت می‌کند
MSEجریمه خطای بزرگواحد مجذور
RMSEجریمه + واحد درستحساس به پرت
R²مقایسه مدل‌ها در دامنه‌های مختلفمی‌تواند گمراه‌کننده باشد
همیشه چند معیار را با هم ببینید. MAE خوب + RMSE بد = مدل روی اکثر داده خوب است ولی روی چند نمونه فاجعه می‌کند.

جمع‌بندی

این مبحث بخشی از دوره آموزش کامل ماشین لرنینگ است.

ابزارهای مرتبط