highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-03-26 17:51:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

avoid boost_score > 1.0

Browse Source
This commit is contained in:
joachim-danswer 2025-02-23 08:29:36 -08:00
parent 625ea6f24b
commit 1954105cc6
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/model_server/custom_models.py
View File

@ -259,7 +259,7 @@ def run_content_classification_inference(
if prob < 0.25:
raw_score = 0.0
elif prob < 0.75:
raw_score = (prob - 0.25) / 0.5
raw_score = min(1.0, (prob - 0.25) / 0.5)
else:
raw_score = 1.0
return (
@ -291,8 +291,8 @@ def run_content_classification_inference(
_prob_to_score(p_temp) for p_temp in output_probabilities_with_temp
]
output_classes = [1] * len(text_inputs)
output_scores = [0.9] * len(text_inputs)
# output_classes = [1] * len(text_inputs)
# output_scores = [0.9] * len(text_inputs)
return [
(predicted_label, output_score)