Skip to content

Commit 8216cf1

Browse files
committed
Fix comment on output format
1 parent f9e53ab commit 8216cf1

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

exercises/01_penguin_classification.ipynb

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -447,8 +447,7 @@
447447
"source": [
448448
"### Task 9: Writing basic training and validation loops\n",
449449
"\n",
450-
"- Before we jump in and write these loops, we must first choose an activation function to apply to the model's outputs. We chose not to include this in the network itself.\n",
451-
" - We need to convert our model outputs into something that can be compared to our targets i.e. `[0, 0, 1]`.\n",
450+
"- Before we jump in and write these loops, we must first choose an activation function to apply to the model's outputs so that they compared to our targets i.e. `[0, 0, 1]`. We chose not to include this in the network itself.\n",
452451
" - Here we are going to use the softmax activation function: see [the PyTorch docs](https://pytorch.org/docs/stable/generated/torch.nn.Softmax.html). It can be seen as a generalization of both the logits and sigmoid functions to handle multi-class classification tasks\n",
453452
" - For those of you who've studied physics, you may be reminded of the partition function in thermodynamics.\n",
454453
" - This activation function is good for classification when the result is one of ``A or B or C``.\n",

0 commit comments

Comments
 (0)