t

functions/model_infer.py

@@ -41,6 +41,6 @@ def predict_from_document(sentences):
     # Set the prediction threshold to 0.8 instead of 0.5, now use mean
     output = (
         prediction.flatten()[: len(sentences)]
-        >= np.mean(prediction) * 1.50  # + np.std(prediction)
+        >= np.mean(prediction) * 1.20  # + np.std(prediction)
     ).astype(int)
     return output, prediction.flatten()[: len(sentences)]