P-NET, Biologically informed deep neural network for prostate cancer classification and discovery
P-NET, Biologically informed deep neural network for prostate cancer classification and discovery
view interactive network architecture
·
·
Biologically informed deep neural network for prostate cancer classification and discovery
To get a local copy up and running, follow these simple steps
Clone the repo
git clone https://github.com/marakeby/pnet_prostate_paper.git
Create conda environment
conda env create --name pnet_env --file=environment.yml
Based on your use, you may need to download one or more of the following
a. Data files (needed to retrain
models and generate figures). Extract the files under _database
directory. If you like to store it somewhere
else, you may need to set the DATA_PATH
variable in config_path.py
accordingly.
b. Log files (needed to
regenerate paper figures). Extract the files under _logs
directory. If you like to store it somewhere else, you
may need to set the LOG_PATH
variable in config_path.py
accordingly.
c. Plots files (a copy of the
paper images). Extract the files under _plots
directory. If you like to store it somewhere else, you may need
to set the PLOTS_PATH
variable in config_path.py
accordingly.
Activate the created conda environment
source activate pnet_env
Add the current diretory to PYTHONPATH, e.g.
export PYTHONPATH=~/pnet_prostate_paper:$PYTHONPATH
To generate all paper figures, run
cd ./analysis
python run_it_all.py
To generate individual paper figure run the different files under the 'analysis' directory, e.g.
cd ./analysis
python figure_1_d_auc_prc.py
For Figure3
, make sure you run prepare_data.py
before running other files
To re-train a model from scratch run
cd ./train
python run_me.py
This will run an experiment 'pnet/onsplit_average_reg_10_tanh_large_testing' which trains a P-NET model on a
training-testing data split of Armenia et al data set and compare it to a simple logistic regression model. The
results of the experiment will be stored under _logs
in a directory with the same name as the experiment.
To run another experiment, you may uncomment one of the lines in the run_me.py to run the corresponding experiment.
Note that some models especially cross validation experiments may be time consuming.
Distributed under the GPL-2.0 License License. See LICENSE
for more information.
Haitham - @HMarakeby
Project Link: https://github.com/marakeby/pnet_prostate_paper
This work was supported in part by the Fund for Innovation in Cancer Informatics, Mark Foundation, Prostate Cancer Foundation, Movember, and the National Cancer Institute at the National Institutes of Health.