This project uses a machine learning model to predict Adverse Drug Reactions (ADRs) based on patient and drug-related features. It is developed as part of a B.Pharm final-year research project to demonstrate the integration of AI in drug safety and pharmacovigilance.
The goal of this project is to train and evaluate a binary classification model to detect the likelihood of an ADR occurring. This project explores how AI can revolutionize pharmacovigilance systems in the healthcare sector.
adr_prediction_model.pkl: Trained machine learning model.adr_prediction.ipynb: Google Colab notebook for training and testing the model.dataset.csv: Cleaned dataset used for training.README.md: Project documentation.
- Upload all files to your Google Colab or local environment.
- Open
adr_prediction.ipynb. - Run the cells to:
- Load dataset
- Preprocess data
- Train model
- Evaluate performance
- The model is saved as
adr_prediction_model.pkl.
A sample synthetic/real dataset was used containing anonymized data such as:
- Age, gender, drug name
- Side effects or symptoms
- Outcome (ADR occurred or not)
- Accuracy: ~50%
- Precision & Recall balanced (Needs improvement)
- Tools for visualization: Confusion matrix, ROC curve
- Python
- Scikit-learn
- Pandas, NumPy, Matplotlib, Seaborn
- Google Colab
This project shows a proof-of-concept for students or researchers in pharmacy, healthcare, or AI domains to understand AI’s impact in ADR prediction.
This project is open for academic and non-commercial use.