fix: embedding_weights.npy file is not found

sangeet2020 / CNN-SHAP-text-classification

A CNN based framework for text classification problem implemented in Keras with a local model explanation using SHAP DeepExplainer class.

1 stars 1 forks source link

fix: embedding_weights.npy file is not found #2

Open neeraj1909 opened 2 years ago

neeraj1909 commented 2 years ago

What I'm trying to achieve

Screenshot from 2021-11-16 15-13-13

sangeet2020 commented 2 years ago

Hi, The repo was not updated since the last changes. I committed some new changes, try it again. In short:

Read me updated: you wont explicitly need embedding_weights.npy. Rather the model now uses pre-trained fastext embeddings.

I used embedding_weights.npy when I was trying to use pytorch implementation however, the training was smooth, SHAP DeepExaplainer produced conflicts with pytorch. So had to switch back to keras. If you wish to explore it, I put it on drive. Link here