diff --git a/02_pytorch_classification.ipynb b/02_pytorch_classification.ipynb index 03064f5d..0a4c44ba 100644 --- a/02_pytorch_classification.ipynb +++ b/02_pytorch_classification.ipynb @@ -1377,7 +1377,7 @@ "\n", " # 1. Forward pass (model outputs raw logits)\n", " y_logits = model_0(X_train).squeeze() # squeeze to remove extra `1` dimensions, this won't work unless model and data are on same device \n", - " y_pred = torch.round(torch.sigmoid(y_logits)) # turn logits -> pred probs -> pred labls\n", + " y_pred = torch.round(torch.sigmoid(y_logits)) # turn logits -> pred probs -> pred labels\n", " \n", " # 2. Calculate loss/accuracy\n", " # loss = loss_fn(torch.sigmoid(y_logits), # Using nn.BCELoss you need torch.sigmoid()\n",