예제 #1
0
    def test_empty_labels(self):
        gt = random.sample(range(10, 60), 50)
        est = random.sample(range(10, 60), 50)

        cm = ConfMatrix(100)
        cm.add(gt, est)
        self.assertEqual(sum(sum(cm.cm)), 50)
예제 #2
0
    def test_simple(self):
        k = 50
        gt = random.sample(range(k), 50)
        est = random.sample(range(k), 50)

        cm = ConfMatrix(k)
        cm.add(gt, est)
        self.assertEqual(sum(sum(cm.cm)), 50)
예제 #3
0
    def test_cut(self):
        gt = [0, 0, 0, 1, 1, 1, 1, 1, 2, 3, 3]
        est = [0, 0, 0, 1, 1, 1, 0, 0, 2, 3, 3]

        cm = ConfMatrix(4, labelset = self.makelabelset(4))
        cm.add(gt, est)
        self.assertEqual(sum(sum(cm.cm[2:, 2:])), 3)
        cm.cut(2)
        self.assertEqual(cm.cm[2, 2], 3)
예제 #4
0
    def test_sort_simple(self):
        gt = [0, 1, 1]
        est = [0, 1, 1]

        cm = ConfMatrix(2, labelset = self.makelabelset(2))
        cm.add(gt, est)
        
        self.assertEqual(cm.cm[1, 1], 2)
        self.assertEqual(cm.labelset[0], 'a')
        
        cm.sort()
        self.assertEqual(cm.cm[0, 0], 2)
        self.assertEqual(cm.labelset[0], 'b')
예제 #5
0
    def test_sort_more_complicated(self):
        gt = [0, 0, 0, 1, 1, 1, 1, 1, 2, 3, 3]
        est = [0, 0, 0, 1, 1, 1, 0, 0, 2, 3, 3]

        cm = ConfMatrix(4, labelset = self.makelabelset(4))
        cm.add(gt, est)
        
        self.assertEqual(cm.cm[1, 1], 3)
        self.assertEqual(cm.cm[3, 3], 2)
        self.assertEqual(cm.cm[1, 0], 2)
        self.assertEqual(cm.labelset, ['a', 'b', 'c', 'd'])

        cm.sort()
        self.assertEqual(cm.cm[0, 0], 3)
        self.assertEqual(cm.cm[2, 2], 2)
        self.assertEqual(cm.cm[0, 1], 2)
        self.assertEqual(cm.labelset, ['b', 'a', 'd', 'c'])