aarya-vsdk commited on
Commit
37ee6be
·
verified ·
1 Parent(s): befd328

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
  "Die Stadt hatte im Jahr 1962 81.400 Einwohner.", # Expected: End of Turn
192
- "Katalanisch ist die am zweitmeisten gesprochene Sprache.", # 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
  "Die Stadt hatte im Jahr 1962 81.400 Einwohner.", # Expected: End of Turn
185
+ "Katalanisch ist die am zweitmeisten gesprochene.", # Expected: Not End of Turn
186
  ]
187
 
188
  for sentence in sentences: