Harmoniczna średnia w funkcji Pythona?
Mam 2 funkcje, które dają precyzję i wyniki przywoływania, muszę utworzyć funkcję średniej harmonicznej zdefiniowaną w tej samej bibliotece, która używa tych dwóch wyników. Funkcje wyglądają tak:
oto funkcje:
def precision(ref, hyp):
"""Calculates precision.
Args:
- ref: a list of 0's and 1's extracted from a reference file
- hyp: a list of 0's and 1's extracted from a hypothesis file
Returns:
- A floating point number indicating the precision of the hypothesis
"""
(n, np, ntp) = (len(ref), 0.0, 0.0)
for i in range(n):
if bool(hyp[i]):
np += 1
if bool(ref[i]):
ntp += 1
return ntp/np
def recall(ref, hyp):
"""Calculates recall.
Args:
- ref: a list of 0's and 1's extracted from a reference file
- hyp: a list of 0's and 1's extracted from a hypothesis file
Returns:
- A floating point number indicating the recall rate of the hypothesis
"""
(n, nt, ntp) = (len(ref), 0.0, 0.0)
for i in range(n):
if bool(ref[i]):
nt += 1
if bool(hyp[i]):
ntp += 1
return ntp/nt
Jak wyglądałaby średnia funkcja harmoniczna? Mam tylko to, ale wiem, że to nie w porządku:
def F1(precision, recall):
(2*precision*recall)/(precision+recall)