KEA with scikit-learn