Невозможно вычислить roc_auc_score, только один класс присутствует в y_true - PullRequest
Новая
       35

Невозможно вычислить roc_auc_score, только один класс присутствует в y_true

0 голосов
/ 20 марта 2020

Я хочу напечатать sklearn roc_auc_score, и у меня есть эта ошибка: 

ValueError: Only one class present in y_true. ROC AUC score is not defined in that case.

Я использую случайный лес для предсказания тем в тексте.

мой код: 

# No validation data in rnd forest
x_train = np.concatenate((x_train, x_valid))
y_train = np.concatenate((y_train, y_valid))

model = RandomForestClassifier(n_estimators=int(clf_params['n_estimators']),
                                   max_features=clf_params['max_features'])
model.fit(pipe_vect.transform(x_train), y_train)

x_test_vect = pipe_vect.transform(x_test)
y_pred = model.predict_proba(x_test_vect) 
auc_score = roc_auc_score(y_test, y_pred)
...