Sketching sklearn API