Путаница матрица и точность теста для PyTorch CNN учебник

Question

Я заинтересован в том, чтобы сообщать только о точности обучения и испытаний, а также о матрице путаницы (скажем, с использованием матрицы смешения sklearn). Как я могу это сделать? В текущем уроке сообщается только о точности обучения и оценки, и мне трудно разобраться, как включить в него код путаницы sklearn. Ссылка на оригинальный учебник здесь: https://github.com/bentrevett/pytorch-sentiment-analysis/blob/master/4%20-%20Convolutional%20Sentiment%20Analysis.ipynb

Answer 1

Очень похоже на функцию binary_accuracy, определенную в руководстве, вы можете реализовать любую метрику c, какую захотите. Все, что вам нужно, - это набор предсказаний модели (в данном случае preds) и истинных целей (y).

Например, для матрицы путаницы вы можете сделать следующее:

from sklearn.metrics import confusion_matrix

def compute_confusion_matrix(preds, y):
    #round predictions to the closest integer
    rounded_preds = torch.round(torch.sigmoid(preds))
    return confusion_matrix(y, rounded_preds)