Update dominoes.ipynb

landoskape · Apr 11, 2024 · 425d8a5 · 425d8a5
1 parent 021c5c6
commit 425d8a5
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/dominoes.ipynb b/dominoes.ipynb
@@ -82,15 +82,31 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 37,
+   "execution_count": 38,
    "id": "94b81df6",
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "dict_keys(['input', 'mask', 'train', 'selection', 'available'])\n"
+      "{'input': tensor([[[0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0.],\n",
+      "         [0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.],\n",
+      "         [0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.],\n",
+      "         [1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.],\n",
+      "         [0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.],\n",
+      "         [0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0.],\n",
+      "         [0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0.]],\n",
+      "\n",
+      "        [[0., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0.],\n",
+      "         [1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.],\n",
+      "         [0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0.],\n",
+      "         [0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.],\n",
+      "         [0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0.],\n",
+      "         [0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.],\n",
+      "         [0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.]]]), 'mask': tensor([[1., 1., 1., 1., 1., 1., 1.],\n",
+      "        [1., 1., 1., 1., 1., 1., 1.]]), 'train': False, 'selection': tensor([[19, 13, 11,  1, 14, 23, 15],\n",
+      "        [10,  2, 26,  8, 23, 13, 12]]), 'available': None}\n"
      ]
     }
    ],
@@ -101,7 +117,7 @@
     "\n",
     "batch = dataset.generate_batch(train=False, batch_size=2, return_full=True)\n",
     "\n",
-    "print(batch.keys())\n",
+    "print(batch)\n",
     "\n",
     "\n",
     "# Figure out how to get the gettarget sorting to work!! Then DominoeDataset is finisheddddddddd\n",