Exemple #1
0
 def train(self, email_adress):
     global all_words, spam_words, probability_spam, count_spams, count_emails
     hemails_with_body = TrainingCorpus(email_adress).hams()
     semails_with_body = TrainingCorpus(email_adress).spams()
     hwords = TrainingCorpus(email_adress).get_words(hemails_with_body)
     swords = TrainingCorpus(email_adress).get_words(semails_with_body)
     all_words = TrainingCorpus(email_adress).all_words(hwords, swords) # all words with their count
     spam_words = TrainingCorpus(email_adress).spam_words(swords) # spam words with their count
     count_spams = TrainingCorpus.count_spams(email_adress) # count of all spam's emails
     count_emails = TrainingCorpus.count_emails(email_adress) # count of all emails
     probability_spam = count_spams / count_emails # probability that email is spam
     pass