aarya-vsdk commited on
Commit
83d7285
·
verified ·
1 Parent(s): e5f1ab3

Quickstart modified

Browse files
Files changed (1) hide show
  1. README.md +0 -7
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
 
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