コード例 #1
0
    def test_get_max_class(self):
        dict1 = {'a':10, 'b':11, 'c': 6}
        dict2 = {'a':14, 'b':11, 'c': 6}

        res1 = BNClassifier.get_max_class(dict1)
        res2 = BNClassifier.get_max_class(dict2)

        self.assertEquals('b', res1)
        self.assertEquals('a', res2)
コード例 #2
0
 def test_counts(self):
     cols = [1,2]
     row = [0,1,0]
     data = [[0,1,1], [0,1,0], [0,0,0], [0,1,1]]
     classes = ['a', 'b', 'a', 'b']
     poss = ['a', 'b']
     counts = BNClassifier.counts(cols, row, data, classes, poss)
     self.assertEquals(1, counts['a'])
     self.assertEquals(2, counts['b'])
コード例 #3
0
    def test_get_list_of_cols(self):
        n1 = BNNode([], [])
        n1.col_index = 0

        n2 = BNNode([], [])
        n2.col_index = 1

        n3 = BNNode([], [])
        n3.col_index = 2

        n1.parents.append(n2)
        n1.parents.append(n3)

        col_list = BNClassifier.get_list_of_cols(n1)
        self.assertEquals(1, col_list[0])
        self.assertEquals(2, col_list[1])
コード例 #4
0
    def test_get_count(self):

        n1 = BNNode([], [])
        n1.col_index = 0

        n2 = BNNode([], [])
        n2.col_index = 1

        n3 = BNNode([], [])
        n3.col_index = 2

        n1.parents.append(n2)
        n1.parents.append(n3)

        cols = [1,2]
        row = [0,1,0]
        data = [[0,1,0], [0,1,0], [0,0,0], [0,1,0]]
        classes = ['a', 'b', 'a', 'b']
        poss = ['a', 'b']
        counts = BNClassifier.get_count(row, n1, data, classes, poss)
コード例 #5
0
    BN.setup_node(node, nodes)

# initial confusion matrix
confusion = {'0':{'0':0, '1':0}, '1':{'0':0, '1':0}}

# do k-fold validation
total_count = 0
kfold = KFold(100, new_data, new_classes)
while kfold.has_next():
    
    dat, cls = kfold.get_next()

    correct_count = 0
    for i in range(0, len(dat)):
        row = dat[i]
        guess = BNClassifier.classify(row, nodes, dat, cls, ['0', '1'])

        if guess == cls[i]:
            correct_count += 1

        confusion[cls[i]][guess] += 1

    total_count += correct_count


for node in nodes:
    pind = ""
    for pnode in node.parents:
        pind += " "+str(pnode.col_index)
    print "Node "+str(node.col_index)+" has parents: "+pind