Is there any way to get variable importance with Keras?

后端 未结 4 1130
渐次进展
渐次进展 2020-12-23 17:51

I am looking for a proper or best way to get variable importance in a Neural Network created with Keras. The way I currently do it is I just take the weights (not the biases

4条回答
  •  心在旅途
    2020-12-23 18:14

    This is a relatively old post with relatively old answers, so I would like to offer another suggestion of using SHAP to determine feature importance for your Keras models. SHAP also allows you to process Keras models using layers requiring 3d input like LSTM and GRU while eli5 cannot.

    To avoid double-posting, I would like to offer my answer to a similar question on Stackoverflow on using SHAP.

提交回复
热议问题