remove sklearn dependency