Я пытаюсь запустить классификатор случайного леса в sklearn:
from sklearn.ensemble import RandomForestClassifier
model_rf_man = RandomForestClassifier(n_estimators= 183, max_features= 'sqrt', max_depth = 10, min_samples_split=5, min_samples_leaf = 3, bootstrap = True)
model_rf_man.fit(x_training_data, y_training_data)
y_pred_rf_man = model_rf_man.predict(x_test_data)
from sklearn.metrics import classification_report
from sklearn.metrics import plot_roc_curve
print(classification_report(y_test_data, y_pred_rf_man))
plot_roc_curve(y_pred_rf_man, x_test_data, y_test_data)
Затем я получаю эту ошибку:
ndarray should be a binary classifier
Первым аргументом, переданным plot_roc_curve
, должен быть ваш объект оценки model_rf_man
. См. документы