Update README.md
Browse files
README.md
CHANGED
|
@@ -137,7 +137,9 @@ prediction = outputs.logits.argmax(dim=-1).item()
|
|
| 137 |
label_map = {0: "NEGATIVE", 1: "POSITIVE"}
|
| 138 |
print(label_map[prediction])
|
| 139 |
|
| 140 |
-
|
|
|
|
|
|
|
| 141 |
Training Data
|
| 142 |
|
| 143 |
Dataset: IMDB Movie Reviews
|
|
|
|
| 137 |
label_map = {0: "NEGATIVE", 1: "POSITIVE"}
|
| 138 |
print(label_map[prediction])
|
| 139 |
|
| 140 |
+
---
|
| 141 |
+
|
| 142 |
+
## Training Details
|
| 143 |
Training Data
|
| 144 |
|
| 145 |
Dataset: IMDB Movie Reviews
|