Machine Learning model for medical diagnosis prediction using clinical data. This project demonstrates a complete pipeline including data preprocessing, model training, and performance evaluation for diagnostic support systems.
- Project Overview
- Features
- Getting Started
- Usage
- Project Structure
- Results
- Contributing
- License
- Contact
This repository contains a machine learning solution for medical diagnosis prediction using:
- Dataset:
medical_data.csv
(clinical parameters and diagnosis results) - Algorithms: Comparison of multiple classification models
- Evaluation: Performance metrics including accuracy, precision, recall, F1-score, and ROC-AUC
✅ Data preprocessing pipeline
✅ Feature engineering and selection
✅ Multiple classifier implementations (Logistic Regression, Random Forest, SVM, etc.)
✅ Model performance comparison
✅ Result visualization (ROC curves, confusion matrices)
✅ Extensible architecture for new algorithms
- Python 3.8+
- Jupyter Notebook
- Virtual environment recommended
# Clone the repository
git clone https://github.com/paratha14/Medical_diagnosis_1.git
cd Medical_diagnosis_1
# Set up virtual environment
python3 -m venv venv
source venv/bin/activate # Windows: venv\Scripts\activate
# Install dependencies
pip install -r requirements.txt
Model | Accuracy | Precision | Recall | F1 Score | AUC-ROC |
---|---|---|---|---|---|
Logistic Regression | 0.89 | 0.88 | 0.87 | 0.87 | 0.93 |
Random Forest | 0.90 | 0.91 | 0.90 | 0.91 | 0.96 |
Support Vector Machine | 0.93 | 0.83 | 0.82 | 0.82 | 0.89 |
- Best Performing Model: Random Forest (92% accuracy)
- Feature Importance: Top 3 features contributing to predictions:
- Blood Glucose Level (24.5%)
- BMI (19.8%)
- Age (15.2%)
Parameter | Value |
---|---|
Total Samples | 15,830 |
Features | 22 |
Positive Cases | 32% |
Missing Values | 0.7% |
Pratham
- GitHub: @paratha14
- Email: prathammohan3@gmail.com
- LinkedIn: https://www.linkedin.com/in/pratham-mohan-47013b2b5?utm_source=share&utm_campaign=share_via&utm_content=profile&utm_medium=android_app
For questions or support requests:
- Open an issue: GitHub Issues
Contributions are welcome! Please:
- Fork the repository
- Create a feature branch (
git checkout -b feature/YourFeature
) - Commit changes (
git commit -m 'Add YourFeature'
) - Push to the branch (
git push origin feature/YourFeature
) - Open a pull request
This project is licensed under the MIT License - see LICENSE for details.