def create_clustering_obj(hit_id, items_map, clustering_list):
"""Creates a clustering object for a clustering
Args:
hit_id (int): Hit #
items_map (dict of tuple to Item): Maps items from occurances in hits
to respective Item class object
clustering_list (list(int)): A clustering of items (represented by
indices of their occuance in hit)
Returns:
Clustering: Object of class clustering
"""
clustering_obj = Clustering()
for cluster in clustering_list:
cluster_obj = Cluster()
for item_id in cluster:
cluster_obj.add_item(items_map[(hit_id, item_id)])
clustering_obj.add_cluster(cluster_obj)
return clustering_obj
