Public release

geom_solver.py

@@ -14,6 +14,10 @@ def my_empty_solution():
 	return np.zeros((20,3)), [(0, 0)]
 
 
+def one_line_solution(n):
+	return np.zeros((n,3)), list(itertools.product(list(range(n)), list(range(n))))
+
+
 def fully_connected_solution(vertices=None):
 	if vertices is None:
 		nverts = 20

main.ipynb

@@ -23,14 +23,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 11,
    "id": "503c6bcb-aa46-46c6-8b86-566b0a470b43",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "adf065a3fb644479b707047372091ea9",
+       "model_id": "bccfd9e961614341bcbd9a4dba7a4a99",
        "version_major": 2,
        "version_minor": 0
       },
@@ -76,7 +76,7 @@
     "\n",
     "from my_solution import predict\n",
     "from helpers import *\n",
-    "from geom_solver import GeomSolver\n",
+    "from geom_solver import GeomSolver, one_line_solution\n",
     "\n",
     "import huggingface_hub  \n",
     "huggingface_hub.login()"
@@ -84,7 +84,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 13,
    "id": "88f4fc8f-efa9-404b-9073-c7d4a73f9075",
    "metadata": {},
    "outputs": [
@@ -94,16 +94,16 @@
      "text": [
       "2.1428452992319946 1.8753585980967309\n",
       "1.2634583203026981 1.2634583203026981\n",
-      "2.542730868722136 1.8969040131019905\n",
-      "1.608507505603034 1.3744402739344028\n",
+      "2.5396091977186415 2.542730868722136\n",
+      "1.5784396320724732 1.3744402739344028\n",
       "1.6604589895130097 1.4945561543255432\n",
       "1.342868026940067 1.5159383672128168\n",
-      "1.3937990493440202 1.492159322757521\n",
-      "1.2228932592065895 1.2545726800817392\n",
+      "1.3318850656966443 1.492159322757521\n",
+      "1.2228932592065895 1.3374845788105776\n",
       "2.1274713572398154 1.9577219841781546\n",
       "1.2389391918918637 1.1907499853854275\n",
       "1.301931556958707 1.5291655623486928\n",
-      "1.4968274710208265 1.349185516938256\n",
+      "1.3586449315580662 1.349185516938256\n",
       "1.300528267625377 1.1364744990514501\n",
       "1.9971234414064012 1.484359575186481\n",
       "1.7864233783834014 1.2997637652539\n",
@@ -111,10 +111,10 @@
       "1.2503083972410651 1.2672496454499096\n",
       "1.9554601999832553 2.4290767592247864\n",
       "1.5169825260067826 1.862860315670855\n",
-      "1.6424954827356966 1.782831036227262\n",
+      "1.6424954827356966 1.887332771285391\n",
       "1.40382235616253 0.9727095077709508\n",
       "2.5531277383538464 2.2956668643388705\n",
-      "1.4201505123873752 1.367457468141674\n",
+      "1.3602630905708255 1.367457468141674\n",
       "1.1352963385173234 1.3060154430256012\n",
       "2.1787116657747303 1.7257473870225462\n",
       "2.199844649970312 2.2529145560735175\n",
@@ -123,7 +123,7 @@
       "1.6472782885605137 1.4709840223700106\n",
       "1.332152878564964 1.6610485499747143\n",
       "Averages\n",
-      "1.6801981329818685 1.6107076453713232\n"
+      "1.6704256833331772 1.6384823283515602\n"
      ]
     }
    ],
@@ -144,15 +144,12 @@
     "    solver = GeomSolver()\n",
     "    vertices, edges = solver.solve(entry)\n",
     "    \n",
-    "    nverts = entry['wf_vertices'].shape[0]\n",
+    "    # nverts = entry['wf_vertices'].shape[0]\n",
     "    # nverts = vertices.shape[0]\n",
     "    nverts = 20\n",
-    "    \n",
-    "    all_verts = list(range(nverts))\n",
-    "    edges0 = list(itertools.product(all_verts, all_verts))\n",
-    "    edges0 = [edg for edg in edges0 if edg[0] < edg[1]]\n",
-    "    \n",
-    "    scores0 = (compute_WED(np.zeros((nverts,3)),\n",
+    "    vertices0, edges0 = one_line_solution(nverts)\n",
+    "\n",
+    "    scores0 = (compute_WED(vertices0,\n",
     "            edges0,\n",
     "            np.array(entry['wf_vertices']),\n",
     "            np.array(entry['wf_edges'])))\n",