sklearn 0.21 版本后无法导入 impute下的IterativeImputer

文件树形的目录

impute
  │  _base.py
  │  _iterative.py
  │  __init__.py

IterativeImputer 是在 _iterative.py

from sklearn.impute import IterativeImputer

会报错。

解决方案：
在 __init__.py 修改为

"""Transformers for missing value imputation"""

from ._base import MissingIndicator, SimpleImputer
from ._iterative import IterativeImputer

__all__ = [
    'MissingIndicator',
    'SimpleImputer',
    'IterativeImputer',
]

这样导入IterativeImputer 不会报错。