Add import statement

This commit is contained in:
Agata 2022-04-24 16:37:18 +02:00
parent 4a8da732fc
commit 72488883fa

View File

@ -32,7 +32,8 @@
"from sklearn.model_selection import train_test_split\n", "from sklearn.model_selection import train_test_split\n",
"from sklearn.metrics import classification_report\n", "from sklearn.metrics import classification_report\n",
"import torch\n", "import torch\n",
"from torch import nn, optim" "from torch import nn, optim\n",
"import torch.nn.functional as F"
] ]
}, },
{ {