Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def word_frequency(training_set, training_label):
- '''
- Fake News Frequency Counter
- '''
- real_frequency_dictionary = {}
- fake_frequency_dictionary = {}
- for i in range(len(training_set)):
- title = training_set[i]
- label = training_label[i]
- words = title.split(" ")
- for word in list(set(words)):
- if label == 1:
- if word not in real_frequency_dictionary:
- real_frequency_dictionary[word] = 1
- else:
- real_frequency_dictionary[word] += 1
- if word not in fake_frequency_dictionary:
- fake_frequency_dictionary[word] = 0
- elif label == 0:
- if word not in fake_frequency_dictionary:
- fake_frequency_dictionary[word] = 1
- else:
- fake_frequency_dictionary[word] += 1
- if word not in real_frequency_dictionary:
- real_frequency_dictionary[word] = 0
- return real_frequency_dictionary, fake_frequency_dictionary
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement