Train and improve a scikit-learn binary classifier | Perplexity Interview Question