多类GridSearch

时间:2017-03-23 04:11:39

标签: scikit-learn

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')

ValueError:y_true中只有一个类。在这种情况下,没有定义ROC AUC分数。

我需要制作一个3级的分类器,但在使用roc_auc的gridsearch中,总会出现一些错误,任何人都可以帮到你吗?

0 个答案:

没有答案