Update README.md
Browse files
README.md
CHANGED
|
@@ -186,7 +186,7 @@ with torch.no_grad():
|
|
| 186 |
# Apply sigmoid to the logits to get probabilities
|
| 187 |
probabilities = torch.sigmoid(outputs.logits).squeeze()
|
| 188 |
|
| 189 |
-
# Determine a threshold for predicting labels
|
| 190 |
threshold = 0.5
|
| 191 |
predicted_label_indices = (probabilities > threshold).nonzero(as_tuple=True)[0]
|
| 192 |
|
|
|
|
| 186 |
# Apply sigmoid to the logits to get probabilities
|
| 187 |
probabilities = torch.sigmoid(outputs.logits).squeeze()
|
| 188 |
|
| 189 |
+
# Determine a threshold for predicting labels
|
| 190 |
threshold = 0.5
|
| 191 |
predicted_label_indices = (probabilities > threshold).nonzero(as_tuple=True)[0]
|
| 192 |
|