Scikit-learn:모델 성능 평가

Sam (토론 | 기여)님의 2022년 2월 14일 (월) 10:29 판 (새 문서: == 개요 == 손쉽게 모델 성능을 평가해보자. 필요한 지표들을 보기 위해선 다음과 같이 짜면 편하다.<syntaxhighlight lang="python"> from sklearn.metri...)
(차이) ← 이전 판 | 최신판 (차이) | 다음 판 → (차이)

개요편집

손쉽게 모델 성능을 평가해보자.


필요한 지표들을 보기 위해선 다음과 같이 짜면 편하다.

from sklearn.metrics import mean_absolute_error, mean_squared_error, mean_squared_log_error, r2_score

def result_evaluate(y_real, y_pred):
    evaluate = {}
    evaluate['mae'] = mean_absolute_error(y_real, y_pred)
    evaluate['rmse'] = np.sqrt(evaluate['mae'])
    evaluate['msle'] = mean_squared_log_error(y_real, y_pred)
    evaluate['rmsle'] = np.sqrt(evaluate['msle'])
    evaluate['r2'] = r2_score(y_real, y_pred)
    
    for i, j in evaluate.items():
        print(i + ' : ' + j)