Skip to content

Commit aeb907e

Browse files
bnestorbnestor
and
bnestor
authored
fixed small error in the global threshold calculation (#221)
Co-authored-by: bnestor <bnestor@github.com>
1 parent 5f97558 commit aeb907e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

InferenceSystem/src/model/fastai_inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def predict(self, wav_file_path):
168168
local_confidences=list(submission['confidence'])
169169
)
170170

171-
result_json['global_prediction'] = int(sum(result_json["local_predictions"]) > self.min_num_positive_calls_threshold)
171+
result_json['global_prediction'] = int(sum(result_json["local_predictions"]) >= self.min_num_positive_calls_threshold)
172172
result_json['global_confidence'] = submission.loc[(submission['confidence'] > self.threshold), 'confidence'].mean()*100
173173
if pd.isnull(result_json["global_confidence"]):
174174
result_json["global_confidence"] = 0

0 commit comments

Comments
 (0)