Skip to content

Commit

Permalink
Add mean matches data
Browse files Browse the repository at this point in the history
  • Loading branch information
vivekjoshy committed Feb 16, 2022
1 parent 4cc6057 commit 9e35974
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions benchmark/split_benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
from typing import Union

import jsonlines
import numpy as np
import trueskill
from prompt_toolkit import print_formatted_text as print, HTML, prompt
from prompt_toolkit.completion import WordCompleter
Expand Down Expand Up @@ -326,6 +327,7 @@ def confident_in_match(match: dict) -> bool:
for line in progress_bar(valid_test_set_matches, total=len(valid_test_set_matches)):
predict_ts_match(match=line)

mean = float(np.array(list(match_count.values())).mean())

print(HTML(f"Confident Matches: <style fg='Yellow'>{confident_matches}</style>"))
print(
Expand Down Expand Up @@ -363,3 +365,4 @@ def confident_in_match(match: dict) -> bool:
)
)
print(HTML(f"Process Duration: <style fg='Yellow'>{ts_time}</style>"))
print(HTML(f"Mean Matches: <style fg='Yellow'>{mean}</style>"))

0 comments on commit 9e35974

Please sign in to comment.