IanRonk commited on
Commit
99542fe
1 Parent(s): 673264c

Make model more conservative

Browse files
Files changed (1) hide show
  1. functions/model_infer.py +1 -1
functions/model_infer.py CHANGED
@@ -38,5 +38,5 @@ def predict_from_document(sentences):
38
  preprop = preprocess(sentences)
39
  prediction = model.predict(preprop)
40
  # Set the prediction threshold to 0.8
41
- output = (prediction.flatten()[: len(sentences)] >= 0.8).astype(int)
42
  return output, prediction.flatten()[: len(sentences)]
 
38
  preprop = preprocess(sentences)
39
  prediction = model.predict(preprop)
40
  # Set the prediction threshold to 0.8
41
+ output = (prediction.flatten()[: len(sentences)] >= 0.85).astype(int)
42
  return output, prediction.flatten()[: len(sentences)]