def draw_function(): df = DecisionFunction() #classifier = nbc.NaiveBayes() classifier = qdc.QuadraticDiscriminantClassifier(2) #classifier = knn.KNN(3) datas = dataset.load ("../datasets/gaussian10k.data") #classifier.load(datas[1000:]) classifier.load(datas[9000:]) classifier.learn() df.draw_classifier(classifier)
def draw_all_roc(): roc = RocCurve() datas = dataset.load ("../datasets/donut1k-20.data") test_set = datas[0:900] # classifier = qdc.QuadraticDiscriminantClassifier(2) # classifier.load(datas[4500:]) # classifier.learn() # roc.draw_classifier(classifier, test_set, "r-") classifier = nbc.NaiveBayes() classifier.load(datas[900:]) classifier.learn() roc.draw_classifier(classifier, test_set, "r-") # test_set = datas[0:900] # # classifier = knn.KNN(3) # classifier.load(datas[900:]) # classifier.learn() # roc.draw_classifier(classifier, test_set, "r-") roc.show()