def main():
print "Creating Train File..."
Data.CreateDataFile("mails-train", "mails-train.txt")
print "Initializing Train File..."
trainingSet = Data.LoadFile("mails-train.txt")
print "Creating Test File..."
Data.CreateDataFile("mails-test", "mails-test.txt")
print "Initializing Test File..."
testSet = Data.LoadFile("mails-test.txt")
print "Extracting Classes.."
classes = Train.ExtractClasses(trainingSet)
print "Training NBC..."
vocabulary, prior, condprob = Train.TrainMultinomialNaiveBayes(
classes, trainingSet)
print "Testing Accuracy..."
percentage = Test.Accuracy(classes, vocabulary, prior, condprob,
testSet)
print "The percentage of correct predictions is ", 100 * percentage, "percent."
print "Get Random Document..."
testDocument = Data.GetDocument(testSet)
print "Do A Time Measurement of the Application of the NBC..."
print "The time is took to do a single application of the NBC on a document is", Test.TimeMeasure(
classes, vocabulary, prior, condprob, testDocument), "seconds."
print "Applying NBC on Document..."
topClass, score = Classify.ApplyMultinomialNaiveBayes(
classes, vocabulary, prior, condprob, testDocument['document'])
def accuracyOnTest(self):
print "Calculating Accuracy..."
print "Creating Test File..."
Data.CreateDataFile(self.testDir, self.folder(self.testDir) + ".txt")
print "Initializing Test File..."
self.testSet = Data.LoadFile(self.folder(self.testDir)+ ".txt")
percentage = Test.Accuracy(self.classes, self.vocabulary, self.prior, self.condprob, self.testSet)
print "The percentage of correct predictions is ",100*percentage,"percent."
print "Done."
