Train GradientBoostingClassifier with 5-Fold Cross-Validation
Final Model Training: GradientBoostingClassifier with 5-Fold CV
Context
Assume the notebook already contains a prepared feature matrix X and a binary target y (0/1), with any necessary preprocessing completed. Your goal is to evaluate and train a final model.
Task
-
Build a scikit-learn Pipeline that uses a GradientBoostingClassifier.
-
Evaluate the model using 5-fold Stratified cross-validation and report the mean ROC-AUC.
-
Fit the model on the full dataset (X, y).
-
Save the trained pipeline to disk as model.pkl.
Deliverable
Provide the full Python code that performs all steps above.
Constraints & Assumptions
-
Preserve the scope, facts, inputs, and requested outputs from the prompt above.
-
If the prompt leaves a detail unspecified, state a reasonable assumption before relying on it.
-
Keep the answer interview-ready: concise enough to present, but concrete enough to implement or evaluate.
Clarifying Questions to Ask
-
Clarify the task, data shape, labels, constraints, and evaluation metric.
-
State assumptions behind the math or modeling technique you choose.
-
Connect theory to practical training, debugging, and deployment implications.
What a Strong Answer Covers
-
Correct definitions and formulas where the prompt requires them.
-
A practical explanation of how the method behaves on real data.
-
Trade-offs, failure modes, diagnostics, and mitigation strategies.
-
Evaluation choices that match the product or modeling objective.
Follow-up Questions
-
How would noisy labels, class imbalance, or distribution shift affect the answer?
-
What would you monitor after deployment?
-
Which baseline would you compare against first?