ismailkonak / ff-algorithm-pytorch-implementation Goto Github PK

 def accuracy_f(self, y_pred, y_true):
        batch_size = y_pred.size(0)
        _, predicted_labels = y_pred.max(1)
        y_pred_onehot = torch.zeros_like(y_pred)
        y_pred_onehot.scatter_(1, predicted_labels.view(-1,1), 1)
        correct = (y_pred_onehot == y_true).sum().item()
        accuracy = correct / (batch_size * y_true.size(1))
        return accuracy