iamgroot42 / permutation_invariant_network

PyTorch implementation for Permutation Invariant Network

Permutation Invariant Network

PyTorch implementation for Permutation Invariant Network. This implementation has been tested to recreate experimental results from the original paper (more information here)

Setup

Clone the repository and install package

git clone https://github.com/iamgroot42/permutation_invariant_network.git
cd permutation_invariant_network
pip install -e .

File Structure

• example.py File with dummy setup to demonstrate use of provided code

• misc.py Helper functions and utilities

• pim.py Model architecture for Permutation Invariant Network

• train.py Functions to train and test meta-classifier