def __init__(self, topic):
self.sieves = []
self.results_dict = dict()
self.results_ordered = []
logger.info('loading topic %s, total mentions: %d', topic.topic_id,
len(topic.mentions))
self.clusters = Clusters(topic.topic_id, topic.mentions)
class RunSystemsSuper(object):
def __init__(self, topic):
self.sieves = []
self.results_dict = dict()
self.results_ordered = []
logger.info('loading topic %s, total mentions: %d', topic.topic_id,
len(topic.mentions))
self.clusters = Clusters(topic.topic_id, topic.mentions)
@staticmethod
def set_sieves_from_config(config, get_rel_extraction):
sieves = []
for _type_tup in config.sieves_order:
sieves.append(
SieveClusterMerger(_type_tup,
get_rel_extraction(_type_tup[0])))
return sieves
def run_deterministic(self):
# pylint: disable=too-many-nested-blocks
for sieve in self.sieves:
start = time.time()
clusters_changed = True
merge_count = 0
while clusters_changed:
clusters_changed = False
clusters_size = len(self.clusters.clusters_list)
for i in range(0, clusters_size):
cluster_i = self.clusters.clusters_list[i]
if cluster_i.merged:
continue
for j in range(i + 1, clusters_size):
cluster_j = self.clusters.clusters_list[j]
if cluster_j.merged:
continue
if cluster_i is not cluster_j:
criterion = sieve.run_sieve(cluster_i, cluster_j)
if criterion:
merge_count += 1
clusters_changed = True
cluster_i.merge_clusters(cluster_j)
cluster_j.merged = True
if clusters_changed:
self.clusters.clean_clusters()
end = time.time()
took = end - start
logger.info(
'Total of %d clusters merged using method: %s, took: %.4f sec',
merge_count, str(sieve.excepted_relation), took)
return self.clusters
def get_results(self):
return self.results_ordered