Commit
•
abb9294
1
Parent(s):
a411c55
update readme
Browse files
README.md
CHANGED
|
@@ -51,4 +51,4 @@ start_prob = torch.max(torch.nn.Softmax(dim=-1)(qa_outputs.start_logits)).item()
|
|
| 51 |
end_prob = torch.max(torch.nn.Softmax(dim=-1)(qa_outputs.end_logits)).item()
|
| 52 |
confidence = (start_prob + end_prob) / 2
|
| 53 |
print(answer, confidence) # 臺灣新北市土城區, 0.92
|
| 54 |
-
```
|
| 51 |
end_prob = torch.max(torch.nn.Softmax(dim=-1)(qa_outputs.end_logits)).item()
|
| 52 |
confidence = (start_prob + end_prob) / 2
|
| 53 |
print(answer, confidence) # 臺灣新北市土城區, 0.92
|
| 54 |
+
```
|