File size: 1,170 Bytes
6ca765f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
# After review with respect to equations
def hierarchical_precision_recall_fmeasure(
true_labels, predicted_labels, ancestors, beta=1.0
):
# Initialize counters for true positives, predicted, and true conditions
true_positive_sum = predicted_sum = true_sum = 0
# Process each instance
for true, predicted in zip(true_labels, predicted_labels):
# Extend the sets with ancestors
extended_true = true.union(
*[ancestors[label] for label in true if label in ancestors]
)
extended_predicted = predicted.union(
*[ancestors[label] for label in predicted if label in ancestors]
)
# Update counters
true_positive_sum += len(extended_true.intersection(extended_predicted))
predicted_sum += len(extended_predicted)
true_sum += len(extended_true)
# Calculate hierarchical precision and recall
hP = true_positive_sum / predicted_sum if predicted_sum else 0
hR = true_positive_sum / true_sum if true_sum else 0
# Calculate hierarchical F-measure
hF = ((beta**2 + 1) * hP * hR) / (beta**2 * hP + hR) if (hP + hR) else 0
return hP, hR, hF
|