【发布时间】:2021-11-21 20:04:02
【问题描述】:
我正在对贷款预测数据使用逻辑回归。我正在使用 GridSearchCV 进行超参数调整,我一直在尝试找到一个可以为 cv 添加多个值的源。例如;我想以 3、5、6、7、10 折运行我的模型。这是我的代码:
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3)
parameters = {'penalty': ['l1', 'l2'],
'C': [0.001, 0.01, 0.1, 1, 10, 100, 1000],
'solver' : ['liblinear', 'newton-cg', 'lbfgs', 'saga', 'sag'],
'multi_class' : ['auto'],
'max_iter' : [5,15,25],
}
s_scaled_X_train = s_scaler.fit_transform(X_train)
s_scaled_X_test = s_scaler.transform(X_test)
logmodel = GridSearchCV(LogisticRegression(), parameters, cv = 10, refit = True)
我试图寻找可以找到解决方案的来源,了解如何在此处的 cv 值中添加更多数字。
logmodel = GridSearchCV(LogisticRegression(), parameters, cv = 10, refit = True)
我尝试了类似的方法:
bb = [3, 5, 6, 7, 10]
cv_folds = bb
我尝试将其添加到日志模型中。
logmodel = GridSearchCV(LogisticRegression(), parameters, cv = cv_folds, refit = True)
这是我运行时遇到的错误
TypeError: cannot unpack non-iterable int object
【问题讨论】:
标签: python machine-learning scikit-learn logistic-regression gridsearchcv