首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >基于K-折叠CV和Gridsearch的支持向量机数据主成分分析与训练

基于K-折叠CV和Gridsearch的支持向量机数据主成分分析与训练
EN

Stack Overflow用户
提问于 2021-07-05 09:25:08
回答 1查看 337关注 0票数 0

我需要训练一个支持向量机模型使用LinearSVC和10倍交叉验证与内部2倍的Gridsearch,以优化伽马和C,但我也必须应用PCA对我的数据,以减少其大小。我应该在模型的简历和训练发生之前还是在循环中应用PCA?在后一种情况下,每个循环都有不同的主成分数,但这是否有缺点呢?

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-07-05 09:46:25

最好的解决方案是创建一个sklearn Pipeline并将两个步骤(PCALinarSvc都放在其中)。这将创建一个实现fit()predict()并可在GridSearchCV中使用的对象。

代码语言:javascript
复制
from sklearn.svm import LinearSVC
from sklearn.pipeline import Pipeline
from sklearn.decomposition import PCA
from sklearn.model_selection import GridSearchCV

pipe = Pipeline([('pca', PCA()),
                 ('clf', LinearSVC())])
params = {
    'pca__n_components' : [2, 5, 10, 15],
    'clf__C' : [0.5, 1, 5, 10],
}

gs = GridSearchCV(estimator=pipe, param_grid=params)
gs.fit(X_train, y_train)
票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/68253660

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档