Advertisement
Guest User

Untitled

a guest
May 25th, 2018
79
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.79 KB | None | 0 0
  1. test_mails_body, test_labels = get_features('test_mails')
  2. test_matrix = vectorizer.transform(test_mails_body)
  3.  
  4. classifier_names = ['Классификатор MultinomialNB','Классификатор GaussianNB','Классификатор BernoulliNB','Классификатор LinearSVC']
  5. classifiers = [MultinomialNB(),GaussianNB(),BernoulliNB(),LinearSVC()]
  6.  
  7.  
  8.  
  9. for name,clf in zip(classifier_names,classifiers):
  10. train_mtr = train_matrix
  11. test_mtr = test_matrix
  12. if 'Gauss' in name:
  13. train_mtr = train_mtr.toarray()
  14. test_mtr = test_mtr.toarray()
  15. clf.fit(train_mtr,mails_labels)
  16. prediction = clf.predict(test_mtr)
  17. print(name)
  18. print(pd.DataFrame(confusion_matrix(test_labels, prediction), columns=['Mails', 'Spam'], index=['Mails', 'Spam']))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement