用Python编写的贝叶斯反垃圾邮件分类:antispam
jopen
9年前
antispam是用Python编写的贝叶斯反垃圾邮件分类。
PyPI: pypi.python.org/pypi/antispam Docs: antispam.readthedocs.org
Installation
pip install antispam
Usage
Use the built-in model provided & trained by author:
import antispam antispam.score("Cheap shoes for sale at DSW shoe store!") # => 0.9657724517163143 antispam.is_spam("Cheap shoes for sale at DSW shoe store!") # => True antispam.score("Hi mark could you please send me a copy of your machine learning homework? thanks") # => 0.0008064840568731558 antispam.is_spam("Hi mark could you please send me a copy of your machine learning homework? thanks") # => False
Train your own modle:
import antispam d = antispam.Detector("my_model.dat") d.train("Super cheap octocats for sale at GitHub.", True) d.train("Hi John, could you please come to my office by 3pm? Ding", False) msg1 = "Cheap shoes for sale at DSW shoe store!" d.score(msg1) # => 0.9999947825633266 d.is_spam(msg1) # => True msg2 = "Hi mark could you please send me a copy of your machine learning homework? thanks" d.score(msg2) # => 4.021280114849398e-08 d.is_spam(msg2) # => False
Save your model:
d.save()