If you are using kaggle, upgrade sklearn version by !pip install -U --upgrade-strategy=eager scikit-learn
!pip install -U --upgrade-strategy=eager scikit-learn