"Scikit-learn:모델 성능 평가"의 두 판 사이의 차이
둘러보기로 가기
검색하러 가기
(새 문서: == 개요 == 손쉽게 모델 성능을 평가해보자. 필요한 지표들을 보기 위해선 다음과 같이 짜면 편하다.<syntaxhighlight lang="python"> from sklearn.metri...) |
(차이 없음)
|
2022년 2월 14일 (월) 10:29 기준 최신판
개요[편집 | 원본 편집]
손쉽게 모델 성능을 평가해보자.
필요한 지표들을 보기 위해선 다음과 같이 짜면 편하다.
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)