Spaces:
Runtime error
Runtime error
| import os | |
| class wording_choice_aggregator(): | |
| """ | |
| A class used for retrieving frequencies based on wording in a message | |
| """ | |
| def get_frequency_of_capatalised_words(self, text): | |
| """ | |
| A function used to retrieve the frequencies of capitalised words in a dataset | |
| :param text: | |
| :return: the frequency of capitalised words in a dataset | |
| """ | |
| number_of_capatalised_words = 0 | |
| for word in text.split(" "): | |
| if word.isupper(): | |
| number_of_capatalised_words = number_of_capatalised_words + 1 | |
| total_number_of_words = len(text.split(" ")) | |
| frequency = number_of_capatalised_words / total_number_of_words | |
| return frequency | |
| def get_frequency_of_violent_or_curse_words(self, text, violent_words_datasets_location): | |
| """ | |
| A function ued for retrieving the frequencies of violent words in a dataset | |
| :param text: | |
| :return: the frequency of violent words in a dataset | |
| """ | |
| dataset_folder = os.path.join(os.getcwd(), violent_words_datasets_location) | |
| list_of_violent_or_curse_words = [] | |
| # Retrieves all words in all of the files in the violent or curse word datasets | |
| for filename in os.listdir(dataset_folder): | |
| with open(os.path.join(dataset_folder, filename), 'r') as file: | |
| for line in file.readlines(): | |
| line = line.strip().replace("\n", " ").replace(",", "") | |
| list_of_violent_or_curse_words.append(line) | |
| number_of_swear_words = 0 | |
| for word in text.split(" "): | |
| if word in list_of_violent_or_curse_words: | |
| number_of_swear_words = number_of_swear_words + 1 | |
| total_number_of_words = len(text.split(" ")) | |
| frequency = number_of_swear_words / total_number_of_words | |
| return frequency | |