SHIROI-07 commited on
Commit
07840b6
·
verified ·
1 Parent(s): 8e1eae2

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -15,7 +15,7 @@ class EmotionModel:
15
  probs = torch.sigmoid(logits)[0]
16
  return {
17
  self.labels[i]: float(probs[i])
18
- for i in range(len(probs)) if probs[i] > 0.3
19
  }
20
 
21
  class SuicidalIntentModel:
 
15
  probs = torch.sigmoid(logits)[0]
16
  return {
17
  self.labels[i]: float(probs[i])
18
+ for i in range(len(probs)) if probs[i] > 0.4
19
  }
20
 
21
  class SuicidalIntentModel: