Building the surrogate model with logistic regression and Scikit-Learn