Fix a crucial bug of mINP calculation!!! (#154)

This commit is contained in:
Jilong Wang
2023-09-04 13:17:50 +08:00
committed by GitHub
parent f280be5a3f
commit 853bb1821d
-13
View File
@@ -128,19 +128,6 @@ def evaluate_rank(distmat, p_lbls, g_lbls, max_rank=50):
all_cmc.append(cmc[:max_rank])
num_valid_p += 1.
# compute average precision
# reference: https://en.wikipedia.org/wiki/Evaluation_measures_(information_retrieval)#Average_precision
num_rel = raw_cmc.sum()
pos_idx = np.where(raw_cmc == 1) # 返回坐标此处raw_cmc为一维矩阵所以返回相当于index
max_pos_idx = np.max(pos_idx)
inp = cmc[max_pos_idx] / (max_pos_idx + 1.0)
all_INP.append(inp)
cmc[cmc > 1] = 1
all_cmc.append(cmc[:max_rank])
num_valid_p += 1.
# compute average precision
# reference: https://en.wikipedia.org/wiki/Evaluation_measures_(information_retrieval)#Average_precision
num_rel = raw_cmc.sum()