Skip to content

bagchi/EIGAN

Repository files navigation

EIGAN

  • docker integration is used to reduce the overhead of setting up environment.
  • users are welcome to use non-docker environments on their own.
  • prepopulated hyperparameters and training logs as well as pretrained models are made available for evaluation.

Docker Setup

To build the docker image

  • replace gpu with cpu in docker-dl-setup/docker-compose.yml in case the system has no gpu
  • script to build the docker image
cd docker-dl-setup
docker-compose build

To run the docker container

./run-docker.sh

To enter the docker container

docker exec -it eigan_devel bash

Training

  • all scripts are run from *.sh files in scripts folder
  • change the hyperparameters, as in example scripts
  • run the scripts inside the docker container
sh scipts/<mimic/mnist/titanic>/<script-name>.sh

Comparison

  • comparison scripts need editing of python scripts
  • replace the names of the pre-populated training histories with the newly generated training histories after training to generate new plots and analysis.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors