summaryrefslogtreecommitdiffstats
path: root/mlscraper/matches.py
diff options
context:
space:
mode:
Diffstat (limited to 'mlscraper/matches.py')
-rw-r--r--mlscraper/matches.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/mlscraper/matches.py b/mlscraper/matches.py
index 3b039d5..d34c430 100644
--- a/mlscraper/matches.py
+++ b/mlscraper/matches.py
@@ -6,6 +6,7 @@ import typing
from functools import cached_property
from itertools import combinations
from itertools import product
+from statistics import mean
from mlscraper.html import get_relative_depth
from mlscraper.html import get_root_node
@@ -135,7 +136,7 @@ class DictMatch(Match):
keys = set(self.match_by_key.keys()).intersection(
set(match.match_by_key.keys())
)
- return sum(
+ return mean(
self.match_by_key[key].get_similarity_to(match.match_by_key[key])
for key in keys
)
@@ -163,7 +164,7 @@ class ListMatch(Match):
def get_similarity_to(self, match: "Match"):
assert isinstance(match, self.__class__)
- return sum(
+ return mean(
lm1.get_similarity_to(lm2)
for lm1, lm2 in product(self.matches, match.matches)
)