def testLocalAuc(self):
testY = numpy.array([-1, -1, 1, 1, 1, 1, 1, -1, -1, 1])
predY = numpy.array([0.987, 0.868, 0.512, 0.114, 0.755, 0.976, 0.05, 0.371, 0.629, 0.819])
self.assertEquals(Evaluator.localAuc(testY, predY, 1.0), Evaluator.auc(predY, testY))
self.assertEquals(Evaluator.localAuc(testY, predY, 0.0), 0)
self.assertEquals(Evaluator.localAuc(testY, testY, 0.2), 1.0)
def testLocalAuc(self):
testY = numpy.array([-1, -1, 1, 1, 1, 1, 1, -1, -1, 1])
predY = numpy.array([
0.987, 0.868, 0.512, 0.114, 0.755, 0.976, 0.05, 0.371, 0.629, 0.819
])
self.assertEquals(Evaluator.localAuc(testY, predY, 1.0),
Evaluator.auc(predY, testY))
self.assertEquals(Evaluator.localAuc(testY, predY, 0.0), 0)
self.assertEquals(Evaluator.localAuc(testY, testY, 0.2), 1.0)
