ML model to classify whether a tumour is benign or malignant using the Wisconsin breast cancer data set.
This model was created using TPOT
The data set can be found on kaggle
This repository contains:
- breast_cancer_notebook.ipynb Open in Colab
- data.csv (full data set from kaggle)
- cleaned_data.csv (cleaned data with top 6 features)
- cleaned_data_pipeline.py (pipeline generated by TPOT w/ minor adjustments)
- pkl_of_cleaned_data_pipeline.pkl (pkl file which can be used for predictions)