Unable to import sklearn.model_selection.StratifiedGroupKFold in Sagemaker Processing job
0
I'm running a Sagemaker Processing job with SKLearnProcessor processor. I tested code locally and it works fine.
from sklearn.model_selection import StratifiedGroupKFold
But when I run it on Sagemaker Processing, it gives me exception:
ImportError: cannot import name 'StratifiedGroupKFold' from 'sklearn.model_selection' (/miniconda3/lib/python3.7/site-packages/sklearn/model_selection/__init__.py)
I verified scikit-learn version on Sagemaker processiong job it shows 0.23.2. which are should have these packages but I am not sure why am I getting this error?