Metadata-Version: 2.1
Name: xswem
Version: 0.1.1
Summary: A simple and explainable deep learning model for NLP.
Home-page: https://github.com/KieranLitschel/XSWEM
Author: Kieran Litschel
Author-email: kieran.litschel@outlook.com
License: MIT
Download-URL: https://github.com/KieranLitschel/XSWEM/tags
Keywords: nlp fast machine learning deep simple tensorflow model word embeddings keras glove explainable swem global local explanations
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.6
Description-Content-Type: text/markdown
Requires-Dist: tensorflow (>=2.0.0)
Requires-Dist: datasets
Requires-Dist: pandas
Requires-Dist: numpy (>=1.16.5)
Requires-Dist: seaborn
Requires-Dist: tqdm

# XSWEM

main [![Build Status](https://travis-ci.org/KieranLitschel/XSWEM.svg?branch=main)](https://travis-ci.org/KieranLitschel/XSWEM), develop [![Build Status](https://travis-ci.org/KieranLitschel/XSWEM.svg?branch=develop)](https://travis-ci.org/KieranLitschel/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](https://arxiv.org/pdf/1805.09843.pdf).

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](https://github.com/KieranLitschel/XSWEM/tree/develop/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](https://github.com/KieranLitschel/XSWEM/blob/main/notebooks/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](https://github.com/KieranLitschel/XSWEM/blob/main/notebooks/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
```


