예제 #1
0
    def testComputeRecall(self):
        trueValues = []
        learnedValues = []
        self.assertEqual(1.0, ModelEvaluation.recall(trueValues,
                                                     learnedValues))

        trueValues = [1]
        learnedValues = []
        self.assertEqual(0.0, ModelEvaluation.recall(trueValues,
                                                     learnedValues))

        trueValues = []
        learnedValues = [1]
        self.assertEqual(1.0, ModelEvaluation.recall(trueValues,
                                                     learnedValues))

        trueValues = [1, 2]
        learnedValues = [2]
        self.assertEqual(0.5, ModelEvaluation.recall(trueValues,
                                                     learnedValues))

        trueValues = [1, 2]
        learnedValues = [2, 3]
        self.assertEqual(0.5, ModelEvaluation.recall(trueValues,
                                                     learnedValues))

        trueValues = [1, 2]
        learnedValues = [2, 3, 4]
        self.assertEqual(0.5, ModelEvaluation.recall(trueValues,
                                                     learnedValues))

        trueValues = [1, 2, 3]
        learnedValues = [2, 3, 4]
        self.assertEqual(2 / 3,
                         ModelEvaluation.recall(trueValues, learnedValues))