2stdev -> stdev

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) + np.std(prediction) * 2
+        >= np.mean(prediction) + np.std(prediction)
     ).astype(int)
     return output, prediction.flatten()[: len(sentences)]