def test_misclass_tied(self): mcs = MisclassSum() assert mcs.rec_count == 0 assert mcs.misclass_sum == 0 refs = CPUTensor([[0]]) preds = CPUTensor([[0.5], [0.5]]) mcs.add(refs, preds) assert mcs.rec_count == 1 assert mcs.misclass_sum == 0 assert mcs.report() == 0 mcs.clear() assert mcs.rec_count == 0 assert mcs.misclass_sum == 0 refs = CPUTensor([[1]]) mcs.add(refs, preds) assert mcs.rec_count == 1 assert mcs.misclass_sum == 1 assert mcs.report() == 1
def test_misclass_sum_report(self): mcs = MisclassSum() assert mcs.rec_count == 0 assert mcs.misclass_sum == 0 refs = CPUTensor([[0, 1, 0, 0]]) preds = CPUTensor([[1, 1, 0, 1]]) mcs.add(refs, preds) assert mcs.rec_count == 4 assert mcs.misclass_sum == 2 assert mcs.report() == 2