WIP switch to scikit-learn-wheel's methods