f4beaac559e00a3676d942dc7e8fea69efc01cfe,catalyst/metrics/tests/test_ndcg.py,,test_sample_ndcg,#,57

Before Change

    outputs = torch.Tensor([y_pred])
    targets = torch.Tensor([y_true])

    true_ndcg_at2 = torch.tensor(1.0 / (1.0 + 1 / math.log2(3)))
    comp_ndcg_at2 = metrics.ndcg(outputs, targets, top_k=[2])[0]

    assert torch.isclose(true_ndcg_at2, comp_ndcg_at2)

After Change

    y_pred1 = [0.5, 0.2, 0.1]
    y_pred2 = [0.5, 0.2, 0.1]
    y_true1 = [1.0, 0.0, 1.0]
    y_true2 = [1.0, 0.0, 1.0]

    outputs = torch.Tensor([y_pred1, y_pred2])
    targets = torch.Tensor([y_true1, y_true2])

    true_ndcg_at2 = 1.0 / (1.0 + 1 / math.log2(3))
    comp_ndcg_at2 = metrics.ndcg(outputs, targets, topk=[2])[0]

    assert np.isclose(true_ndcg_at2, comp_ndcg_at2)

Instances