Quickstart modified
Browse files
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
|