-
Notifications
You must be signed in to change notification settings - Fork 10
Open
Labels
enhancementNew feature or requestNew feature or request
Milestone
Description
@shntnu as per our discussion.
An starting point could be this function:
copairs/src/copairs/compute.py
Lines 89 to 96 in fc829c0
| def average_precision(rel_k) -> np.ndarray: | |
| """Compute average precision based on binary list sorted by relevance""" | |
| tp = np.cumsum(rel_k, axis=1) | |
| num_pos = tp[:, -1] | |
| k = np.arange(1, rel_k.shape[1] + 1) | |
| pr_k = tp / k | |
| ap = (pr_k * rel_k).sum(axis=1) / num_pos | |
| return ap |
shntnu
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request