[英]Why doesn't fit_transform work in this sklearn Pipeline example?
我是sklearn Pipeline的新手,并遵循示例代码。 我在其他示例中看到我们可以执行pipeline.fit_transform(train_X)
,因此我在此处的pipeline.fit_transform(X)
上对管道进行了同样的尝试,但它给了我一个错误
“ return self.fit(X,** fit_params).transform(X)
TypeError:fit()恰好接受3个参数(给定2个)“
如果删除svm部分并将管道定义为pipeline = Pipeline([("features", combined_features)])
,我仍然会看到错误。
有谁知道fit_transform
为什么在这里不起作用?
from sklearn.pipeline import Pipeline, FeatureUnion
from sklearn.grid_search import GridSearchCV
from sklearn.svm import SVC
from sklearn.datasets import load_iris
from sklearn.decomposition import PCA
from sklearn.feature_selection import SelectKBest
iris = load_iris()
X, y = iris.data, iris.target
# This dataset is way to high-dimensional. Better do PCA:
pca = PCA(n_components=2)
# Maybe some original features where good, too?
selection = SelectKBest(k=1)
# Build estimator from PCA and Univariate selection:
combined_features = FeatureUnion([("pca", pca), ("univ_select", selection)])
# Use combined features to transform dataset:
X_features = combined_features.fit(X, y).transform(X)
svm = SVC(kernel="linear")
# Do grid search over k, n_components and C:
pipeline = Pipeline([("features", combined_features), ("svm", svm)])
param_grid = dict(features__pca__n_components=[1, 2, 3],
features__univ_select__k=[1, 2],
svm__C=[0.1, 1, 10])
grid_search = GridSearchCV(pipeline, param_grid=param_grid, verbose=10)
grid_search.fit(X, y)
print(grid_search.best_estimator_)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.