Merge pull request #3 from k0nakona/fixes

fix-of-shape
andriygav · Oct 3, 2024 · f150980 · f150980
2 parents 44f2979 + cf9a570
commit f150980
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sem8/main.ipynb b/sem8/main.ipynb
@@ -264,8 +264,8 @@
     "X_train_pt = torch.from_numpy(X_train).float()\n",
     "X_test_pt = torch.from_numpy(X_test).float()\n",
     "\n",
-    "y_train_pt = torch.from_numpy(y_train).float()\n",
-    "y_test_pt = torch.from_numpy(y_test).float()\n",
+    "y_train_pt = torch.from_numpy(y_train).float().view(-1,1)\n",
+    "y_test_pt = torch.from_numpy(y_test).float().view(-1,1)\n",
     "\n",
     "train_dataset = torch.utils.data.TensorDataset(X_train_pt, y_train_pt)\n",
     "test_dataset = torch.utils.data.TensorDataset(X_test_pt, y_test_pt)"