Nice, I think this is a good way to reuse sklearn stuff.