Advertisement
Guest User

Untitled

a guest
Mar 19th, 2018
76
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.09 KB | None | 0 0
  1. def word_frequency(training_set, training_label):
  2. '''
  3. Fake News Frequency Counter
  4. '''
  5. real_frequency_dictionary = {}
  6. fake_frequency_dictionary = {}
  7.  
  8. for i in range(len(training_set)):
  9. title = training_set[i]
  10. label = training_label[i]
  11. words = title.split(" ")
  12. for word in list(set(words)):
  13. if label == 1:
  14. if word not in real_frequency_dictionary:
  15. real_frequency_dictionary[word] = 1
  16. else:
  17. real_frequency_dictionary[word] += 1
  18.  
  19. if word not in fake_frequency_dictionary:
  20. fake_frequency_dictionary[word] = 0
  21.  
  22. elif label == 0:
  23. if word not in fake_frequency_dictionary:
  24. fake_frequency_dictionary[word] = 1
  25. else:
  26. fake_frequency_dictionary[word] += 1
  27.  
  28. if word not in real_frequency_dictionary:
  29. real_frequency_dictionary[word] = 0
  30.  
  31. return real_frequency_dictionary, fake_frequency_dictionary
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement