annaritz / cancer-drug-pred

Source code for Cancer Drug Prediction based on Genetic Mutations

cancer-drug-pred

This repo contains the source code for Cancer Drug Prediction based on Genetic Mutations

1. Download the .py file and open in a Python IDE, using Python 3.6
2. Dowload Ananconda, specifically scikit-learn 0.19.1, pandas 0.20.3, and pydot 1.2.3
3. Download the given datafiles
4. Run the program.
5. The program will output a decision tree structure in .png format, a drug prediction accuracy value as a decimal between 0 and 1, and a drug prediction for a randomly selected patient.

Note: cv_results_ may need to be replaced with grid_scores_ in older versions of pandas.