erikhenriksson
commited on
Commit
•
60d800a
1
Parent(s):
22bf4b8
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 |
|