def on_epoch_end(self, net, dataset_train, dataset_valid, **kwargs):
EpochScoring.on_epoch_end(self, net, dataset_train, dataset_valid)
X_test, y_test = data_from_dataset(dataset_valid)
y_pred = net.predict(X_test)
cm = confusion_matrix(y_test, y_pred)
history = net.history
history.record("confusion_matrix", cm)
def on_epoch_end(self, net, dataset_train, dataset_valid, **kwargs):
EpochScoring.on_epoch_end(self, net, dataset_train, dataset_valid)
X_test, y_test = data_from_dataset(dataset_valid)
y_pred = net.predict(X_test)
cm = confusion_matrix(y_test, y_pred)
sample_num = np.sum(cm, axis=1)
sample_num = np.tile(sample_num, (sample_num.size, 1))
cm = cm * (1 / np.transpose(sample_num))
cm = np.round(cm * 100)
cm1 = cm.astype(np.int8)
history = net.history
history.record("confusion_matrix", cm1)
