import sys from BayesFilter import BayesFilter import BayesDB db = BayesDB.createOrLoad('out') print db.describe()
import sys import BayesDB classifier = sys.argv[1].strip() if classifier == 'NONE': classifier = None training_data = open(sys.argv[2], 'r') db_file = sys.argv[3] db = BayesDB.createOrLoad(db_file) docs = 0 for line in training_data: line = line.strip() ## assume non-trivial sanitisation done upstream if line == "": continue db.addDoc(line, classifier) docs+=1 print 'Classified %d docs as %s' % (docs, classifier) print 'Writing DB file %s ...' % db_file db.dump() print 'New DB state:' db.describe() print 'Done'
import sys from BayesFilter import BayesFilter import BayesDB db = BayesDB.createOrLoad(sys.argv[1]) filter = BayesFilter(db) print filter.predict(sys.argv[2], sys.argv[3])