__author__ = "techbk" import os from pool import Pool DClasses = ["clinton", "lawyer", "math", "medical", "music", "sex"] base = "learn/" p = Pool() for classname in DClasses: p.learn(base + classname, classname) base = "test/" for classname in DClasses: dir = os.listdir(base + classname) for file in dir: res = p.probability(base + classname + "/" + file) print(classname + ": " + file + ": " + str(res))