문돌이 존버/데이터 분석
2021. 7. 30.
(Explainable AI) SHAP 그래프 해석하기! feat. 실전 코드
이번 시간엔 파이썬 라이브러리로 구현된 SHAP을 직접 써보며 그 결과를 이해해보겠습니다. 보스턴 주택 데이터셋을 활용해보겠습니다. import pandas as pd import numpy as np # xgb 모델 사용 from xgboost import XGBRegressor, plot_importance from sklearn.model_selection import train_test_split import shap X, y = shap.datasets.boston() X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=2021) model = XGBRegressor() model.fit(X_t..
