A machine learning project for classifying breast tumors as malignant or benign using the Breast Cancer Wisconsin dataset available in Scikit-learn.
This project demonstrates a complete machine learning workflow, including:
- Loading the Breast Cancer dataset
- Creating a Pandas DataFrame for data exploration
- Exploratory Data Analysis (EDA)
- Data visualization using scatter plots
- Correlation analysis with a heatmap
- Feature selection based on correlation
- Model training and evaluation
The project uses the Breast Cancer Wisconsin Diagnostic Dataset provided by Scikit-learn through the load_breast_cancer() function.
Dataset characteristics:
- 569 samples
- 30 numerical features
- Binary classification
- Malignant
- Benign
- Python
- Scikit-learn
- NumPy
- Pandas
- Matplotlib
- Seaborn
Clone the repository:
git clone https://github.com/python-is-life2022/breast-cancer-classification.git
cd breast-cancer-classificationInstall the required packages:
pip install -r requirements.txtOpen the Jupyter Notebook or Google Colab notebook and run the cells in order.
- Load the dataset
- Explore the dataset
- Visualize the data
- Analyze feature correlations
- Select important features
- Train the machine learning model
- Evaluate the model
breast-cancer-classification/
│
├── breast_cancer_classification.ipynb
├── requirements.txt
├── README.md
This project is intended for educational purposes.