A simple and explainable deep learning model for NLP.
Project description
XSWEM
A simple and explainable deep learning model for NLP implemented in TensorFlow.
Based on SWEM-max as proposed by Shen et al. in Baseline Needs More Love: On Simple Word-Embedding-Based Models and Associated Pooling Mechanisms, 2018.
This package is currently in development. The purpose of this package is to make it easy to train and explain SWEM-max.
You can find demos of the functionality we have implemented in the notebooks directory of the package. Each notebook has a badge that allows you to run it yourself in Google Colab. We will add more notebooks as new functionality is added.
For a demo of how to train a basic SWEM-max model see train_xswem.ipynb.
So far we have implemented the global explainability method proposed in section 4.1.1 of the original paper. You can see a demo of this method in the notebook global_explain_embedding_components.ipynb.
We are currently implementing some methods for local explainability.
How to install
This package can be installed using pip.
pip install xswem
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.