Skip to content
GANs for Precision Oncology
Python Jupyter Notebook
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

Adversarial Training for Precision Oncology


This repo showcases two generative adversarial net (GAN) approaches to precision oncology. In each folder the scripts and notebooks to train a model and to apply a trained model are provided.


To install the necessary libraries run:

pip install -r requirements.txt


With a dataset of co-occurring gene pairs, train a GAN to learn the pair distribution in order to generate and discriminate co-occurring gene pairs.


With two dataset consisting of a patients disease, demographics and genetic information and the corresponding treatments train a conditional GAN to produce suitable treatment suggestions based on the patient information.


This work is based on the medGAN approach introduced in the following paper:

Generating Multi-label Discrete Patient Records using Generative Adversarial Networks
Edward Choi, Siddharth Biswal, Bradley Malin, Jon Duke, Walter F. Stewart, Jimeng Sun  
Machine Learning for Healthcare (MLHC) 2017
You can’t perform that action at this time.