Add main script to map tags (WIP)

woctezuma · Jul 22, 2023 · 8f0fed9 · 8f0fed9
1 parent 7ab346a
commit 8f0fed9
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/map_tags.py b/map_tags.py
@@ -0,0 +1,18 @@
+from src.data_utils import POLL_DICTIONARY_FNAME, POLL_SUMMARY_FNAME
+from src.json_utils import load_json
+from src.matrix_utils import get_adjacency_matrix, get_tag_joint_game_matrix
+from src.tag_utils import sort_tags
+
+
+def main():
+    games = load_json(POLL_SUMMARY_FNAME)
+    tag_dictionary = load_json(POLL_DICTIONARY_FNAME)
+
+    tags, tag_names = sort_tags(tag_dictionary)
+
+    tag_joint_game_matrix = get_tag_joint_game_matrix(games, tags)
+    tags_adjacency_matrix = get_adjacency_matrix(tag_joint_game_matrix)
+
+
+if __name__ == "__main__":
+    main()