Update README.md
Browse files
README.md
CHANGED
|
@@ -72,6 +72,10 @@ with torch.no_grad():
|
|
| 72 |
probs = torch.nn.functional.softmax(outputs, dim=1)
|
| 73 |
predicted_class = probs.argmax().item()
|
| 74 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 75 |
print("Predicted class:", predicted_class)
|
| 76 |
print("Probabilities:", probs.numpy())
|
| 77 |
|
|
|
|
| 72 |
probs = torch.nn.functional.softmax(outputs, dim=1)
|
| 73 |
predicted_class = probs.argmax().item()
|
| 74 |
|
| 75 |
+
|
| 76 |
+
# 0 = Negative, 1 = Positive
|
| 77 |
+
|
| 78 |
+
|
| 79 |
print("Predicted class:", predicted_class)
|
| 80 |
print("Probabilities:", probs.numpy())
|
| 81 |
|