Deep-Sets

A modified implementation of the Set Transformer using Tensorflow and Keras.

Problem Definition

A function is permutation equivariant iff for any permutation :

Similarly, a function is permutation equivariant if:

Neural networks tend to satisfy neither of these conditions by construction. This repo implements a version of the Set Transformer.

Initial Setup

Install dependencies

This repo requires Python 3.7 and above. Install requirements by running:

pip install -r requirements.txt

Then expoert src to path:

export PYTHONPATH=PATH_TO_REPO/src

To test the scripts, run pytest in the root directory