aarya-vsdk commited on
Commit
d373254
·
verified ·
1 Parent(s): 474be0f

Quickstart modified

Browse files
Files changed (1) hide show
  1. README.md +1 -8
README.md CHANGED
@@ -166,13 +166,6 @@ class TurnDetector:
166
 
167
  probabilities = self._softmax(logits[0])
168
  predicted_label = np.argmax(probabilities)
169
- print(f"Logits: {logits[0]}")
170
- print(f"Probabilities:")
171
- print(f" Not End of Turn: {probabilities[0]:.1%}")
172
- print(f" End of Turn: {probabilities[1]:.1%}")
173
- print(f"Predicted Label: {predicted_label}")
174
- print(f"Confidence: {probabilities[predicted_label]:.1%}")
175
- print(f"Confidence (class=1): {np.max(probabilities):.1%}")
176
  confidence = float(np.max(probabilities))
177
 
178
  return predicted_label, confidence
@@ -189,7 +182,7 @@ if __name__ == "__main__":
189
 
190
  sentences = [
191
  "Некоторые птицы предпочитают гнездиться на определенных островах.", # Expected: End of Turn
192
- "Иоаннами были названы монархи из 16 стран.", # Expected: Not End of Turn
193
  ]
194
 
195
  for sentence in sentences:
 
166
 
167
  probabilities = self._softmax(logits[0])
168
  predicted_label = np.argmax(probabilities)
 
 
 
 
 
 
 
169
  confidence = float(np.max(probabilities))
170
 
171
  return predicted_label, confidence
 
182
 
183
  sentences = [
184
  "Некоторые птицы предпочитают гнездиться на определенных островах.", # Expected: End of Turn
185
+ "Иоаннами были названы монархи из.", # Expected: Not End of Turn
186
  ]
187
 
188
  for sentence in sentences: