Update dominoes.ipynb

dominoes.ipynb

@@ -114,7 +114,13 @@
    "source": [
     "# TODO For refactoring\n",
     "\n",
+    "# integrate \".to(device)\" into the dataset classes for the generate_batch method\n",
+    "\n",
+    "\n",
     "# start working on TSP dataset\n",
+    "# -- finish reward method for TSP dataset\n",
+    "# -- need to think more carefully about how to handle the permutation problem etc etc\n",
+    "\n",
     "# start working on supervised learning child of parent dataset class (and make dominoeDataset an child of that also!)"
    ]
   },
@@ -130,12 +136,13 @@
     "\n",
     "from time import time\n",
     "import torch\n",
+    "from dominoes.datasets import support\n",
     "from dominoes.datasets import DominoeDataset"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 2,
    "id": "94b81df6",
    "metadata": {},
    "outputs": [
@@ -144,9 +151,9 @@
      "output_type": "stream",
      "text": [
       "tensor([[1., 1., 0., 0., 0., 0., 0., 0., 0.],\n",
-      "        [1., 1., 1., 1., 1., 1., 0., 0., 0.],\n",
-      "        [1., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
-      "        [1., 1., 1., 1., 1., 0., 0., 0., 0.]])\n"
+      "        [1., 1., 1., 1., 1., 1., 1., 0., 0.],\n",
+      "        [1., 1., 1., 1., 1., 1., 1., 0., 0.],\n",
+      "        [1., 1., 0., 0., 0., 0., 0., 0., 0.]])\n"
      ]
     }
    ],