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

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

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

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

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

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