Mizuiro-sakura
commited on
Commit
•
7296856
1
Parent(s):
da95118
Update README.md
Browse files
README.md
CHANGED
@@ -61,7 +61,7 @@ results = model(torch.tensor(X1).to(torch.int64),torch.tensor(X2).to(torch.int64
|
|
61 |
|
62 |
# 最も高い値のインデックスを取得する
|
63 |
max_result=torch.argmax(results.logits)
|
64 |
-
print(max_result)
|
65 |
```
|
66 |
# モデルの精度 accuracy of model
|
67 |
79.80339588918764
|
|
|
61 |
|
62 |
# 最も高い値のインデックスを取得する
|
63 |
max_result=torch.argmax(results.logits)
|
64 |
+
print(max_result+1)
|
65 |
```
|
66 |
# モデルの精度 accuracy of model
|
67 |
79.80339588918764
|