diff --git a/typeConsolidation/TypeConsolidation.py b/typeConsolidation/TypeConsolidation.py index 0d83805..3f52ff8 100644 --- a/typeConsolidation/TypeConsolidation.py +++ b/typeConsolidation/TypeConsolidation.py @@ -222,11 +222,13 @@ def optimization(mention, candidateTypes, type2score, disjointContr, hierarchica print('\t\tloading occurrence.....') sys.stdout.flush() pearsonValues = {} - +big_universes = {'camphalfbloodroleplay', 'fallout', 'enmarveldatabase', + 'forgottenrealms', 'harrypotter', 'leagueoflegends', 'lego', + 'psychology', 'starwars', 'villains', 'wowwiki', 'xenoblade'} for univ in universes: print('\t\t' + univ) - sys.stdout.flush - if univ == 'wowwiki' or univ == "enmarveldatabase" or univ == "starwars" or univ == "harrypotter": + sys.stdout.flush() + if univ in big_universes: continue; occurrenceFile = basedir + univ + "/type-occurrence" occurrences = {}