6027fdf7aa77c038bf533ebcf7b704b1c79c3bc4,model/metric.py,,my_metric,#,4

Before Change

def my_metric(y_input, y_target):
    assert len(y_input) == len(y_target)
    correct = 0
    for y0, y1 in zip(y_input, y_target):
        if np.array_equal(y0, y1):
            correct += 1
    return correct / len(y_input)


def my_metric2(y_input, y_target):

After Change

    pred = torch.argmax(output, dim=1)
    assert pred.shape[0] == len(target)
    correct = 0
    correct += torch.sum(pred == target).item()
    return correct / len(target)

def my_metric2(output, target, k=3):
    pred = torch.topk(output, k, dim=1)[1]

Instances