def main(): #Simple argument consistency check #Exit if arguments not specified if len(sys.argv) != 3: print("usage: python3 nblearn.py TRAININGFILE MODELFILE") sys.exit(0) #Create an instance of Naive Bayes class in training mode naive_bayes = NB('TRAIN') #Fit the data with the training file naive_bayes.fit(sys.argv[1]) #Train the classifier naive_bayes.train() #Output a model file to the file specified naive_bayes.generate_model(sys.argv[2])