Mizuiro-sakura commited on
Commit
60e557e
1 Parent(s): 411e41a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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