Sklearn:TransformerMixin的check_estimator相当于什么?

时间:2018-02-19 13:57:58

标签: python python-3.x machine-learning scikit-learn feature-extraction

我正在尝试使用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

0 个答案:

没有答案