回答編集履歴
1
補足追記
test
CHANGED
|
@@ -1 +1,3 @@
|
|
|
1
1
|
`y_pred = y_pred.int()` でTrueは1に、Falseは0に変換されます。numpyでのastype(int)と同様の動作です。
|
|
2
|
+
|
|
3
|
+
floatにしたいのであれば、`y_pred = y_pred.float()`です。
|