Skip to content

PyTorch implementation with sklearn interface of a GAN based imputer

Notifications You must be signed in to change notification settings

utkarshp/GAINImputer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GAINImputer

PyTorch implementation with sklearn interface of GAIN: Missing Data Imputation using Generative Adversarial Nets by Jinsung Yoon, James Jordan, Mihaela van der Schaar.

This class is based on the original PyTorch implementation of the code available here: https://github.com/dhanajitb/GAIN-Pytorch/tree/master, which in turn is an adaptation of the original Tensorflow implementation by the authors here: https://github.com/jsyoon0823/GAIN.

Reference: Yoon, J., Jordon, J. & Schaar, M.. (2018). GAIN: Missing Data Imputation using Generative Adversarial Nets. Proceedings of the 35th International Conference on Machine Learning, in Proceedings of Machine Learning Research 80:5689-5698 Available from https://proceedings.mlr.press/v80/yoon18a.html.

About

PyTorch implementation with sklearn interface of a GAN based imputer

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages