Update spacy/cli/find_threshold.py

Co-authored-by: Sofie Van Landeghem <svlandeg@users.noreply.github.com>
This commit is contained in:
Raphael Mitsch 2022-11-17 10:48:10 +01:00 committed by GitHub
parent ba857c6958
commit 5500a58c00
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -207,7 +207,7 @@ def find_threshold(
else: else:
if not silent: if not silent:
print( print(
f"\nBest threshold: {round(best_threshold, ndigits=4)} with value of {scores[best_threshold]}." f"\nBest threshold: {round(best_threshold, ndigits=4)} with {scores_key} value of {scores[best_threshold]}."
) )
return best_threshold, scores[best_threshold], scores return best_threshold, scores[best_threshold], scores