b1d71998e8dad7604a3eb3ff0279fb275b8ae7e2,catalyst/dl/tests/test_ppv_tpr_f1.py,,test_meter,#,67

Before Change

    assert counts_dict["tp"] == batch_size
    assert counts_dict["fp"] == 0 and counts_dict["fn"] == 0, \
        "There should be no fp and fn for this test case."
    ppv, tpr, f1 = meter.value()
    ppv, tpr, f1 = map(lambda x: round(x, 3), [ppv, tpr, f1])
    assert ppv == tpr == f1 == 1, \
        "No fp and fn means that all metrics should be =1."

After Change

    runs_tests_on_meter_counts_and_value(meter, num_tp_check=batch_size)

    // testing with seg; shape (batch_size, n_channels, h, w)
    meter.reset()
    batch_size = 16
    binary_y, binary_pred = create_dummy_tensors_seg(batch_size)
    meter.add(binary_pred, binary_y)
    runs_tests_on_meter_counts_and_value(meter, num_tp_check=batch_size*15*15)

Instances