Scikit learn 多类网格搜索

ValueError:y_中只有一个类为true。在这种情况下，ROC AUC分数没有定义。

---

我需要制作一个具有3类的分类器，但是在使用

roc_auc

的gridsearch中，总会出现一些错误，任何人都可以帮助您？

您使用的cv=5将在内部使用一个简单的KFold，其中可能会出现训练折叠仅由单个类组成的情况。您应该使用分层方法来维持交叉验证拆分中的类平衡。看见还显示了一些数据示例。您使用的cv=5将在内部使用一个简单的KFold，在该KFold中，可能只包含一个类。您应该使用分层方法来维持交叉验证拆分中的类平衡。看见还显示了一些数据示例

self.train_label_df = pd.DataFrame(train_label)
self.test_label_df = pd.DataFrame(test_label)

data = self.train_df.values

label = self.train_label_df.values

(68914, 249)

(68914, 3)

clf = RandomForestClassifier()

param_grid = {'n_estimators': [100],'criterion': ['entropy'],'max_depth': [10] }

gs = GridSearchCV(estimator=clf, param_grid=param_grid, cv=5, scoring='roc_auc')