【发布时间】:2020-12-22 16:16:24
【问题描述】:
我正在制作一个由 tfidf 矢量化器和 xgboost 分类器组成的管道,并且我正在尝试为我的问题找到矢量化器的最佳参数。但是,我收到以下错误:
Cannot clone object Text2TfIdfTransformer(max_df=0.5, max_features=1000), as the constructor either does not set or modifies parameter max_df.
代码如下:
class Text2TfIdfTransformer(BaseEstimator):
def __init__(self, max_df = 1, max_features = 3000):
self._model = TfidfVectorizer(max_df, max_features, sublinear_tf=True)
pass
def fit(self, data, df_y=None):
self._model.fit(data)
return self
def transform(self, text):
return self._model.transform(text)
pl_xgb_tf_idf = Pipeline(steps=[('tfidf',Text2TfIdfTransformer()),
('xgboost', XGBClassifier(objective='multi:softmax'))])
parameters = {'tfidf__max_df':[.5,.6], 'tfidf__max_features': [1000]}
grid = GridSearchCV(pl_xgb_tf_idf, param_grid=parameters, cv=5)
grid.fit(X,labels)
我不确定在调用 init 时是否应该声明变量 max_df 和 max_features 但如果我不在这里声明它们,我会得到另一个错误(估计器没有任何变量)
我确定我缺少一些基本的东西,但我找不到它到底是什么,任何帮助将不胜感激!
如有重要信息遗漏,请追问!
【问题讨论】:
标签: python scikit-learn pipeline gridsearchcv