时间戳

首页 > 解决方案 > sklearn中的GridSearchCV如何选择交叉验证集？

machine-learning - sklearn中的GridSearchCV如何选择交叉验证集？

问题描述

svc = GridSearchCV(SVC(), param_grid,scoring='f1_macro', verbose=1000)
svc.fit(X_train, y_train)
predictions = svc.predict(X_test)

我多次运行此代码，但结果是相同的。GridSearchCV 是否选择相同的交叉验证集（不是随机选择）？

标签： machine-learningscikit-learngridsearchcv

解决方案

GridSearchCV默认情况下使用KFold交叉验证器，并且默认情况下KFold不打乱数据。要启用洗牌，您必须这样做

from sklearn.model_selection import KFold

cross_validator = KFold(shuffle=True)
svc = GridSearchCV(SVC(), param_grid, scoring='f1_macro', 
                   verbose=1000, cv=cross_validator)
svc.fit(X_train, y_train)
predictions = svc.predict(X_test)

推荐阅读