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