Fix: use scikit-learn instead of sklearn

Fix: use scikit-learn instead of sklearn #336