diff --git a/code/absinth.py b/code/absinth.py index 1938a3808c9b8e9ecb5f141b4c625b0a7141738a..5a67ed27bc421700ff939e885c9dc83cb0d19bb2 100644 --- a/code/absinth.py +++ b/code/absinth.py @@ -1,4 +1,5 @@ import os # for reading files +import sys #from tqdm import tqdm # for counting seconds import spacy # for nlp import networkx as nx # for visualisation @@ -186,7 +187,7 @@ if __name__ == '__main__': corpus_path = '/proj/absinth/wikipedia.txt.dump.20140615-en.SZTAKI' G = Graph() #initialises graph - G.build(corpus_path, 'division_by_zero', filters) #builds graph from corpus with target and filters + G.build(corpus_path, sys.argv[1], filters) #builds graph from corpus with target and filters for hub in G.root_hubs(): print(hub.key,'-->', list(hub.neighbors.keys()), '\n') #prints senses