def testNoWinningLabels(self): """Inferring 0/4 classes should return 0 winning labels.""" model = ClassificationModel() inferenceResult = numpy.array([0, 0, 0, 0]) topLabels = model.getWinningLabels(inferenceResult) self.assertFalse(topLabels)
def testNoWinningLabels(self): """Inferring 0/4 classes should return 0 winning labels.""" model = ClassificationModel() inferenceResult = numpy.array([0, 0, 0, 0]) topLabels = model.getWinningLabels(inferenceResult) self.assertFalse(topLabels)
def testWinningLabels(self): """ Tests whether classification base class returns multiple labels correctly. """ model = ClassificationModel() inferenceResult = numpy.array([3, 1, 4, 0, 1, 0]) topLabels = model.getWinningLabels(inferenceResult, numLabels=1) self.assertTrue(numpy.allclose(topLabels, numpy.array([2])), "Output should be label 2.") topLabels = model.getWinningLabels(inferenceResult, numLabels=2) self.assertTrue(numpy.allclose(topLabels, numpy.array([2, 0])), "Output should be labels 2 and 0.") # Test only nonzero labels are returned. inferenceResult = numpy.array([3, 0, 4, 0, 0, 0]) topLabels = model.getWinningLabels(inferenceResult, numLabels=5) self.assertTrue(numpy.allclose(topLabels, numpy.array([2, 0])), "Output should be labels 2 and 0.")
def testWinningLabels(self): """ Tests whether classification base class returns multiple labels correctly. """ model = ClassificationModel() inferenceResult = numpy.array([3, 1, 4, 0, 1, 0]) topLabels = model.getWinningLabels(inferenceResult, numLabels=1) self.assertTrue(numpy.allclose(topLabels, numpy.array([2])), "Output should be label 2.") topLabels = model.getWinningLabels(inferenceResult, numLabels=2) self.assertTrue(numpy.allclose(topLabels, numpy.array([2, 0])), "Output should be labels 2 and 0.") # Test only nonzero labels are returned. inferenceResult = numpy.array([3, 0, 4, 0, 0, 0]) topLabels = model.getWinningLabels(inferenceResult, numLabels=5) self.assertTrue(numpy.allclose(topLabels, numpy.array([2, 0])), "Output should be labels 2 and 0.")