This project tackled the challenge of predicting the probabilities of individuals receiving H1N1 and seasonal influenza vaccinations using machine learning techniques. Leveraging data from the 2009 National H1N1 Flu Survey (NHFS) conducted by CDC, this project aimed to inform public health strategies and improve vaccine uptake rates. Demonstrating its effectiveness, the project secured 2nd place at AnalyticaX, a machine learning competition organized by IIT Indore during Fluxus'24.
Model Used: BaggingClassifier with XGBoostClassifier as the base estimator.
Key Points:-
1. Tackled the feature columns' missing values.
2. Addressed the taget variables' class imbalance.
3. Two separate models are instantiated using BaggingClassifier with XGBoostClassifier as the base estimator for each target variable.
4. Implemented hyperparameter tuning using Stratified RandomizedSearchCV and Stratified GridSearchCV.
5. The best hyperparameters are directly used in the code.
6.Utilizes ROC AUC for evaluation within each model. Additionally, we calculated an overall ROC AUC by combining both models' predictions, providing a comprehensive assessment of performance.
Results:-
The mean ROC AUC scores for each target variable and the overall score are summarized below:
-> H1N1 vaccine: Mean ROC AUC score = 0.8415
-> Seasonal vaccine: Mean ROC AUC score = 0.8588
-> Overall ROC AUC score = 0.8501
29 Feb 2024
Trusted by 63100+ Generalists. Try it now, free to use
Start making more money