Commit
•
f462261
1
Parent(s):
632b5e1
Update README.md
Browse files
README.md
CHANGED
|
@@ -68,7 +68,7 @@ results = model(torch.tensor(X1).to(torch.int64),torch.tensor(X2).to(torch.int64
|
|
| 68 |
|
| 69 |
# 最も高い値のインデックスを取得する
|
| 70 |
max_result=torch.argmax(results.logits)
|
| 71 |
-
print(max_result)
|
| 72 |
```
|
| 73 |
|
| 74 |
|
|
|
|
| 68 |
|
| 69 |
# 最も高い値のインデックスを取得する
|
| 70 |
max_result=torch.argmax(results.logits)
|
| 71 |
+
print(max_result+1)
|
| 72 |
```
|
| 73 |
|
| 74 |
|