Esempio n. 1
0
def main(args):
    try:
        opts, args = getopt.getopt(args, "hd:p:", ["help", "database=", "pickle="])
    except getopt.GetoptError as msg:
        usage(msg)
        return 1
    if len(args) != 1:
        usage()
        return 1
    cdbname = args[0]
    dbname = usedb = None
    for opt, arg in opts:
        if opt in ("-h", "--help"):
            usage()
            return 0
    dbname, usedb = storage.database_type(opts)
    store = storage.open_storage(dbname, usedb)
    bayes = CdbClassifier()
    items = []
    for word in store._wordinfokeys():
        record = store._wordinfoget(word)
        prob = store.probability(record)
        items.append((word, str(prob)))
    cdbfile = open(cdbname, "wb")
    cdb.cdb_make(cdbfile, items)
    cdbfile.close()
Esempio n. 2
0
 def store(self):
     items = [(self.statekey, "%d,%d" % (self.nham, self.nspam))]
     for word, wi in self.wordinfo.items():
         if isinstance(word, str):
             word = word.encode("utf-8")
         items.append((word, "%d,%d" % (wi.hamcount, wi.spamcount)))
     db = open(self.db_name, "wb")
     cdb.cdb_make(db, items)
     db.close()
Esempio n. 3
0
 def save_wordinfo(self, db_file):

        items = []

        for word, record in self.wordinfo.items():

            prob = Classifier.probability(self, record)

            items.append((word, str(prob)))

        cdb.cdb_make(db_file, items)
Esempio n. 4
0
    store = storage.open_storage(dbname, usedb)

    bayes = CdbClassifier()

    items = []

    for word in store._wordinfokeys():

        record = store._wordinfoget(word)

        prob = store.probability(record)

        items.append((word, str(prob)))

    cdbfile = open(cdbname, "wb")

    cdb.cdb_make(cdbfile, items)

    cdbfile.close()
 if __name__ == "__main__":

    sys.exit(main(sys.argv[1:]))

 if __name__ == "__main__":

    sys.exit(main(sys.argv[1:]))



Esempio n. 5
0
 def save_wordinfo(self, db_file):
     items = []
     for word, record in self.wordinfo.iteritems():
         prob = Classifier.probability(self, record)
         items.append((word, str(prob)))
     cdb.cdb_make(db_file, items)
Esempio n. 6
0
        usage(msg)
        return 1

    if len(args) != 1:
        usage()
        return 1
    cdbname = args[0]

    dbname = usedb = None
    for opt, arg in opts:
        if opt in ("-h", "--help"):
            usage()
            return 0

    dbname, usedb = storage.database_type(opts)
    store = storage.open_storage(dbname, usedb)

    bayes = CdbClassifier()
    items = []
    for word in store._wordinfokeys():
        record = store._wordinfoget(word)
        prob = store.probability(record)
        items.append((word, str(prob)))
    cdbfile = open(cdbname, "wb")
    cdb.cdb_make(cdbfile, items)
    cdbfile.close()


if __name__ == "__main__":
    sys.exit(main(sys.argv[1:]))