Iam-invisible/CancerPrediction_Model

Predicting-cancer-type

Classifying the tumors and non-tumor gene expressions into 34 different types: 33 cancer types + 1 normal sample type.

1. Installation
2. How to Run
3. References

Installation

Using data from the paper by Mostavi, et al.

git clone git@github.com:comedi-team3/CancerPrediction_Model.git
cd CancerPrediction_Model

How to Run

Running locally

python main.py

Running through Docker

docker build -t cancer_prediction .
docker run -it --name cancerprediction_model -v ~/CancerPrediction_Model:/workspace cancer_prediction /bin/bash
cd Model
python main.py

References

• Mostavi, Milad, et al. Convolutional neural network models for cancer type prediction based on gene expression. BMC Medical Genomics 13 (2020): 1-13.