Skip to content

Latest commit

 

History

History
12 lines (5 loc) · 929 Bytes

File metadata and controls

12 lines (5 loc) · 929 Bytes

Feature-Perturbation-Augmentation

This repository contains the code to reproduce the results of our paper Feature Perturbation Augmentation (FPA): arXiv.
Requirements for running the notebooks are PyTorch, PyTorch Lightning and Numba.

We provide code for training the models with FPA, but pre-trained weights can also be downloaded here and have to be put in a folder called "weights".

As a first step, importance estimates are generated using the get_estimators.ipynb notebook and the perturbation-based evaluation can then be performed using the perturb.ipynb notebook.