【发布时间】:2017-12-11 22:54:24
【问题描述】:
我是初学者,下面有以下代码。
from sklearn.naive_bayes import GaussianNB
from sklearn.decomposition import PCA
pca = PCA()
model = GaussianNB()
steps = [('pca', pca), ('model', model)]
pipeline = Pipeline(steps)
cv = StratifiedShuffleSplit(n_splits=5, test_size=0.2, random_state=42)
modelwithpca = GridSearchCV(pipeline, param_grid= ,cv=cv)
modelwithpca.fit(X_train,y_train)
这是一个本地测试,我想要完成的是,
我。对数据集执行 PCA
二。仅使用默认参数使用高斯朴素贝叶斯
三。使用 StratifiedShuffleSplit
所以最后我希望将上述步骤转移到另一个转储分类器、数据集和特征列表的函数中,以测试性能。
dump_classifier_and_data(modelwithpca, dataset, features)
在 param_grid 部分,我不想测试任何参数列表。如果有意义的话,我只想在高斯朴素贝叶斯中使用默认参数。我要改变什么?
关于我如何实例化分类器对象也应该有任何改变吗?
【问题讨论】:
标签: python machine-learning scikit-learn sklearn-pandas grid-search