def kfcvb(mi, k=10): correct = [] tested = [] c = 0 t = 0 for i in range(k): #mark_test_set(mi, k, i) MessageFeatures.test_fold = i MessageFeatures.folds = k bn = BinomialNB(mi) bn.train() bn.test_marked() c = bn.correct t = bn.t correct.append(c) tested.append(t) print (1.0*sum(correct)/sum(correct))
def binomial(mi): MessageFeatures.test_fold = -1 classifier = BinomialNB(mi) classifier.train() classifier.test()