update to use latest scikit-learn API