Implementing a user-based system with scikit-learn