Skip to content

Latest commit

 

History

History
63 lines (44 loc) · 3.13 KB

File metadata and controls

63 lines (44 loc) · 3.13 KB

Breast-Cancer-Survival-Prediction

Breast Cancer Survival Prediction with Clinical and Gene Expression Data

To see demo video click here

Overview

Breast cancer is the second most common cancer after skin cancer in women. It affects nearly 2.1 million women every year globally. Modern genetic measurement technologies such as Next Generation Sequencing and Microarray show light on active and inactive genes responsible for certain biological functions. Comparing these gene expression changes for healthy and cancerous cells sheds light on the causal factors associated with cancer growth. This gives better insights into cancer prognosis and treatment plans such as therapy.

Machine learning helps us to learn complex patterns from intricate data that are difficult for human beings to recognize manually. One such use case is estimating the survival time and hence preventing unnecessary surgical treatments. Here in this project, the survival status is predicted using gene expression data and clinical data using classical machine learning algorithms.

Dataset

The dataset going to be used in the case study is published in Nature Communications (Pereira et al., 2016), also available in Kaggle named as Molecular Taxonomy of Breast Cancer International Consortium (METABRIC) database. This is part of a Canada-UK Project which contains targeted sequencing data of 1,980 primary breast cancer samples. The associated clinical and genomic data was downloaded from cBioPortal. The dataset was originally collected by Professor Carlos Caldas from Cambridge Research Institute and Professor Sam Aparicio from the British Columbia Cancer Center in Canada.

Flask Deployment

The deployment.zip file contains python code corresponding to a deployed version of the project as flask based webapp.

Python Version

Python 3.9.12

Setting up virtual environment

# Installing Virtual Environment
python -m pip install --user virtualenv

# Creating New Virtual Environment
python -m venv envname

# Activating Virtual Environment
source envname/bin/activate

# Upgrade PIP
python -m pip install --upgrade pip

# Installing Packages
python -m pip install -r requirements.txt

How to run

# Download Dataset
python run.py -s d

# Model Training
python run.py -s m

# Predict on Test Data
python run.py -s t

Results

Performance on test data

Model Accuracy AUC Sensitivity Specificity FPR
Decision Tree 0.740876 0.737327 0.704918 0.769737 0.230263
Support Vector 0.583942 0.532787 0.065574 1.000000 0.000000
Random Forest 0.722628 0.704702 0.540984 0.868421 0.131579