This repository complements the paper Missing Data Imputation using Optimal Transport (Muzellec B., Josse J., Boyer C., Cuturi, M.):

• experiment.py allows to reproduce the imputation benchmark therein;
• imputers.py contains the classes corresponding to algorithms 1 and 3;
• data_loaders.py contains data loading utilities for the UCI ML repository datasets on which experiments are run;
• utils.py contains methods of general utility, and the implementation of MAR and MNAR missing data mechanisms in particular;
• softimpute.py contains the implementation of the softimpute baseline.

An example notebook is also available: UCI_demo.ipynb.

Muzellec B., Josse J., Boyer C., Cuturi, M.: Missing Data Imputation using Optimal Transport

• Python 3+
• PyTorch
• GeomLoss
• POT

To use the data loading utilities in data_loaders.py, wget is also required.