我正在尝试使用FunctionTransformer
创建自定义功能提取器供个人使用。我希望尽可能使其与scikit-learn API兼容,以避免出现警告和错误,但我无法通过check_estimator
测试。
但是,我注意到,尽管是DictVectorizer
的孩子,但提供的功能提取器BaseEstimator
仍未通过测试。我的理解是TransformerMixin
似乎导致了这个问题。
from sklearn.utils.estimator_checks import check_estimator
from sklearn.svm import LinearSVC
check_estimator(LinearSVC) # passes
from sklearn.feature_extraction import DictVectorizer
check_estimator(DictVectorizer) # fails
是否有与check_estimator
同等的TransformerMixin
?