Машинное обучение - проблема классификации - PullRequest
1 голос
/ 17 марта 2020

Существует ли какая-либо команда для вызова ошибочно классифицированных значений при прогнозировании любых значений в модели, кратко те, которые были классифицированы неточно.

1 Ответ

0 голосов
/ 18 апреля 2020

В API scikit-learn такого метода нет. Однако вы можете написать это в несколько строк:

import numpy as np

X_train, X_test, y_train, y_test = ...  # some dataset
model = ...  # some scikit learn model
model.fit(X_train, y_train)
y_pred = model.predict(X_test)

# command to get misclassified examples assuming you have numpy arrays
X_test[np.argwhere(y_pred != y_test)]
...