Add import statement

2022-04-24 16:37:18 +02:00 · 2022-04-24 16:37:18 +02:00 · 72488883fa
commit 72488883fa
parent 4a8da732fc
1 changed files with 2 additions and 1 deletions
--- a/classification_net.ipynb
+++ b/classification_net.ipynb
@ -32,7 +32,8 @@
    "from sklearn.model_selection import train_test_split\n",
    "from sklearn.metrics import classification_report\n",
    "import torch\n",
-    "from torch import nn, optim"
+    "from torch import nn, optim\n",
+    "import torch.nn.functional as F"
   ]
  },
  {