remove scikit-learn dependency