From f9460a429d7da07537c23eaaec13b5452366c86e Mon Sep 17 00:00:00 2001 From: frederik Date: Thu, 15 Feb 2024 15:20:03 +0100 Subject: [PATCH] fix: priv_thresh is a percentage, not a fraction --- scripts/server.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/scripts/server.py b/scripts/server.py index 5704fc7..729eb29 100755 --- a/scripts/server.py +++ b/scripts/server.py @@ -223,12 +223,10 @@ def predict(sample, sensitivity): # Sort by score p_sorted = sorted(p_labels.items(), key=operator.itemgetter(1), reverse=True) - log.debug("DATABASE SIZE: %d", len(p_sorted)) - log.debug("HUMAN-CUTOFF AT: %d", int(len(p_sorted)*priv_thresh)/10) -# -# # Remove species that are on blacklist + human_cutoff = max(10, int(len(p_sorted) * priv_thresh / 100.0)) - human_cutoff = max(10, int(len(p_sorted) * priv_thresh)) + log.debug("DATABASE SIZE: %d", len(p_sorted)) + log.debug("HUMAN-CUTOFF AT: %d", human_cutoff) for i in range(min(10, len(p_sorted))): if p_sorted[i][0] == 'Human_Human':