blob: c1f2065821223b3ba9dbc96d561e7dfbdbce51dd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
"""
A rudimentary way to train additional spam into our pickle file.
Example usage:
$ python train_spam.py /path/to/spam.pkl
blah blah blah^D
"""
import sys
from ConfigParser import SafeConfigParser as ConfigParser, \
NoSectionError, NoOptionError
from spambayes.storage import PickledClassifier
config = ConfigParser()
with open("/etc/email_gateway.cfg") as fp:
config.readfp(fp)
def main():
pickle_filename = sys.argv[-1]
bayes = PickledClassifier(pickle_filename)
message = sys.stdin.readlines()
bayes.learn(message, True)
bayes.store()
if __name__ == '__main__':
main()
|