Skip to content

This project uses the Naive Bayes algorithm to classify breast cancer using the scikit-learn library. It includes data preprocessing, model training, evaluation metrics, and visualizations.

Notifications You must be signed in to change notification settings

wizardoftrap/Breast-Cancer-Classification-using-Naive-Bayes

Repository files navigation

Breast Cancer Classification using Naive Bayes

The goal of this project is to predict whether a tumor is malignant or benign based on various features extracted from breast cancer diagnostic data. The dataset used is the Breast Cancer Wisconsin Diagnostic dataset provided by scikit-learn.


Project Overview

  • Dataset: Breast Cancer Wisconsin Diagnostic (from scikit-learn)
  • Algorithm: Gaussian Naive Bayes
  • Metrics: Accuracy, Confusion Matrix, Classification Report, ROC Curve

Results

Confusion Matrix

The confusion matrix shows the true vs. predicted classifications:

Confusion Matrix


ROC Curve

The ROC Curve visualizes the model's ability to distinguish between classes:

ROC Curve


Technologies Used

  • Programming Language: Python 3.x
  • Libraries:
    • Scikit-learn
    • Matplotlib
    • Seaborn
    • NumPy

COLAB Link

https://colab.research.google.com/drive/1u4EnQU0eEsK_7UeFrrOUd8MqA9xYO9nA?usp=sharing

About

This project uses the Naive Bayes algorithm to classify breast cancer using the scikit-learn library. It includes data preprocessing, model training, evaluation metrics, and visualizations.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages