[Python, shap] InvalidModelError: Model type not yet supported by TreeExplainer: <class 'sklearn.svm._classes.SVR'>

 

 

SVR(Support Vector Regressor) 모델에 대해 SHAP(SHapley Additive exPlanations)을 구하려다 발생한 에러. 

 

사용한 SVR 모델은 다음과 같다. 

# SVR 모델
from sklearn.svm import SVR
from sklearn.metrics import mean_squared_error

sv_reg = SVR(kernel = "linear")
sv_reg.fit(train_x,train_y)
sv_pred = sv_reg.predict(test_x)

 

 

찾아보니 SVR은 tree기반 모델이 아니라서 TreeExplainer을 사용하면 안된다고한다. 

 

# shap value
import shap

# SVM은 Tree기반이 아니라서 TreeExplainer 사용 불가
explainer = shap.KernelExplainer(sv_reg.predict, train_x)
shap_values = explainer.shap_values(test_x) # Shap Values 계산

 

TreeExplainer대신에 KernelExplainer을 사용하여 에러 해결.

KernelExplainer는 TreeExplainer보다 속도가 느려서 이점 염두해 사용해야 할 거 같다.