aws / aws/sagemaker-scikit-learn-extension
No module named 'sklearn.preprocessing.label'
- Dominant language
- Python
- Stars
- 41
- Forks
- 34
- PR merge metrics
- No merged PRs in 30d
Description
Using scikit-learn=='0.24.1' on SageMaker, getting the following error:
```
---------------------------------------------------------------------------
ModuleNotFoundError Traceback (most recent call last)
in
51 # from sklearn.pipeline import Pipeline
52
---> 53 from sagemaker_sklearn_extension.preprocessing import RobustLabelEncoder
54 from sagemaker_sklearn_extension.preprocessing import RobustStandardScaler
55 from sagemaker_sklearn_extension.preprocessing import ThresholdOneHotEncoder
~/SageMaker/custom-miniconda/miniconda/envs/ef-datascience/lib/python3.7/site-packages/sagemaker_sklearn_extension/preprocessing/__init__.py in
20 from .data import QuadraticFeatures
21 from .data import RobustStandardScaler
---> 22 from .encoders import NALabelEncoder
23 from .encoders import RobustLabelEncoder
24 from .encoders import RobustOrdinalEncoder
~/SageMaker/custom-miniconda/miniconda/envs/ef-datascience/lib/python3.7/site-packages/sagemaker_sklearn_extension/preprocessing/encoders.py in
20 from sklearn.base import BaseEstimator, TransformerMixin
21 from sklearn.preprocessing import LabelEncoder, OneHotEncoder, OrdinalEncoder, KBinsDiscretizer
---> 22 from sklearn.preprocessing.label import _encode, _encode_check_unknown
23 from sklearn.utils.validation import check_is_fitted, column_or_1d, _num_samples, check_array, check_X_y
24 from sagemaker_sklearn_extension.impute import RobustImputer
ModuleNotFoundError: No module named 'sklearn.preprocessing.label'
```
Contributor guide
Assessment
This issue has not been assessed yet.