maximuspowers commited on
Commit
e34c9e0
1 Parent(s): d1224d5

change threshold

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def predict_ner_tags(sentence):
30
  outputs = model(input_ids=input_ids, attention_mask=attention_mask)
31
  logits = outputs.logits
32
  probabilities = torch.sigmoid(logits)
33
- predicted_labels = (probabilities > 0.6).int() # remember to try your own threshold
34
 
35
  result = []
36
  tokens = tokenizer.convert_ids_to_tokens(input_ids[0])
 
30
  outputs = model(input_ids=input_ids, attention_mask=attention_mask)
31
  logits = outputs.logits
32
  probabilities = torch.sigmoid(logits)
33
+ predicted_labels = (probabilities > 0.5).int() # remember to try your own threshold
34
 
35
  result = []
36
  tokens = tokenizer.convert_ids_to_tokens(input_ids[0])