alihan/HiRAG
1
0
Fork 0
mirror of https://github.com/hhy-huang/HiRAG.git synced 2025-09-16 23:52:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

[fixbug] fix: division by zero while computing change rate of cluster sparsity

Browse Source
This commit is contained in:
haoyuhuang
2025-05-07 18:26:06 +08:00
parent 4bc684d3c8
commit 4a22510ad8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
hirag/_cluster_utils.py
View File

@@ -219,7 +219,7 @@ class Hierarchical_Clustering(ClusteringAlgorithm):
cluster_sizes = Counter(np.concatenate(clusters))
# calculate cluster sparsity
cluster_sparsity = 1 - sum([x * (x - 1) for x in cluster_sizes.values()])/(len(nodes) * (len(nodes) - 1))
cluster_sparsity_change_rate = (abs(cluster_sparsity - pre_cluster_sparsity) / pre_cluster_sparsity)
cluster_sparsity_change_rate = (abs(cluster_sparsity - pre_cluster_sparsity) / (pre_cluster_sparsity + 1e-8))
pre_cluster_sparsity = cluster_sparsity
logging.info(f"[Cluster Sparsity: {round(cluster_sparsity, 4) * 100}%]")
# stop if there will be no improvements on clustering