Commit 8d0de831 authored by Noah Brackenbury's avatar Noah Brackenbury

Improved TestWebMessageModel

parent 6d662f06
...@@ -62,6 +62,7 @@ def WebMessage_testModel(self): ...@@ -62,6 +62,7 @@ def WebMessage_testModel(self):
word_relevance = (language_arrays[language][word])/(list(language_arrays[language].values())[0]) word_relevance = (language_arrays[language][word])/(list(language_arrays[language].values())[0])
language_relevance[language] = language_relevance[language] + word_relevance language_relevance[language] = language_relevance[language] + word_relevance
message_language = max(language_relevance, key=language_relevance.get) message_language = max(language_relevance, key=language_relevance.get)
if message_language != "en":
suggested_subject_list.append(message_language) suggested_subject_list.append(message_language)
# clean up text for analysis # clean up text for analysis
...@@ -93,6 +94,7 @@ def WebMessage_testModel(self): ...@@ -93,6 +94,7 @@ def WebMessage_testModel(self):
suggested_tags_set = set(suggested_subject_list) suggested_tags_set = set(suggested_subject_list)
correct_tags += len(suggested_tags_set.intersection(message_tags_set)) / len(message_tags_set) correct_tags += len(suggested_tags_set.intersection(message_tags_set)) / len(message_tags_set)
if len(suggested_tags_set) != 0:
excess_tags += len(suggested_tags_set.difference(message_tags_set)) / len(suggested_tags_set) excess_tags += len(suggested_tags_set.difference(message_tags_set)) / len(suggested_tags_set)
correct_language = True correct_language = True
...@@ -112,10 +114,10 @@ def WebMessage_testModel(self): ...@@ -112,10 +114,10 @@ def WebMessage_testModel(self):
if not suggested_tags_set.intersection(sr) and not suggested_tags_set.intersection(so): if not suggested_tags_set.intersection(sr) and not suggested_tags_set.intersection(so):
type_accuracy += 1 type_accuracy += 1
correct_tags /= len(test_messages) correct_tags = float(correct_tags) / float(len(test_messages))
excess_tags /= len(test_messages) excess_tags = float(excess_tags) / float(len(test_messages))
language_accuracy /= len(test_messages) language_accuracy = float(language_accuracy) / float(len(test_messages))
type_accuracy /= len(test_messages) type_accuracy = float(type_accuracy) / float(len(test_messages))
end_time = time.time() end_time = time.time()
uptime = end_time - start_time uptime = end_time - start_time
human_uptime = str(datetime.timedelta(seconds=int(uptime))) human_uptime = str(datetime.timedelta(seconds=int(uptime)))
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment