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