def test_auc_report_probs(self):
auc = AUC()
refs = CPUTensor([[0, 0, 1, 1]])
preds = CPUTensor([[0.1, 0.4, 0.35, 0.8]])
auc.add(refs, preds)
assert auc.report() == .75
def test_auc_report_probs(self):
auc = AUC()
refs = CPUTensor([[0, 0, 1, 1]])
preds = CPUTensor([[0.1, 0.4, 0.35, 0.8]])
auc.add(refs, preds)
assert auc.report() == .75
def test_auc_report_binary(self):
auc = AUC()
refs = CPUTensor([[0, 1, 0, 0]])
preds = CPUTensor([[1, 1, 0, 1]])
auc.add(refs, preds)
assert auc.report() == (2.0 / 3.0)
def test_auc_report_binary(self):
auc = AUC()
refs = CPUTensor([[0, 1, 0, 0]])
preds = CPUTensor([[1, 1, 0, 1]])
auc.add(refs, preds)
assert auc.report() == (2.0 / 3.0)
