chris/fastbook
1
0
Fork 0
mirror of https://github.com/fastai/fastbook.git synced 2025-04-05 18:30:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update 06_multicat.ipynb (#349)

Browse Source 
Fix binary_cross_entropy
This commit is contained in:
Joe Bender 2020-12-11 16:36:58 -05:00 committed by GitHub
parent 08746b427e
commit 1cf5630742
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
06_multicat.ipynb
View File

@ -976,7 +976,7 @@
"source": [ "source": [
"def binary_cross_entropy(inputs, targets):\n", "def binary_cross_entropy(inputs, targets):\n",
" inputs = inputs.sigmoid()\n", " inputs = inputs.sigmoid()\n",
" return -torch.where(targets==1, inputs, 1-inputs).log().mean()" " return -torch.where(targets==1, 1-inputs, inputs).log().mean()"
] ]
}, },
{ {