fix d_effective figure

EleutherAI · Oct 30, 2023 · c4de538 · c4de538
1 parent f0c460e
commit c4de538
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 7 deletions.
diff --git a/analysis/hf_reanalysis/D_effective-raw.html b/analysis/hf_reanalysis/D_effective-raw.html
diff --git a/analysis/hf_reanalysis/reanalysis.ipynb b/analysis/hf_reanalysis/reanalysis.ipynb
@@ -1100,28 +1100,44 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 19,
+   "execution_count": 22,
    "id": "a2a488a9-d86e-439c-ac1a-4e12fa96bdd8",
    "metadata": {
     "tags": []
    },
    "outputs": [],
    "source": [
-    "runs = {\n",
-    "    **runs, \n",
-    "    'D_effective': compute_effective_data(best_result.x, runs['L'], runs['N'])\n",
+    "runs_early_stopped = {\n",
+    "    **runs_early_stopped, \n",
+    "    'D_effective': compute_effective_data(best_result.x, runs_early_stopped['L'], runs_early_stopped['N'])\n",
     "}\n",
     "\n",
     "scaling_scatter(\n",
-    "    runs,\n",
+    "    runs_early_stopped,\n",
     "    x_key='D',\n",
     "    y_key='R',\n",
     "    z_key='N',\n",
     "    color_key='D_effective',\n",
-    "    color_type='log',\n",
+    "    color_type='log10',\n",
     "    savepath='D_effective-raw.html',\n",
     ")"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "a5c98894-763d-4d35-a17a-23ebadd82ae7",
+   "metadata": {},
+   "source": [
+    "## Functional Form of $D_E$"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "411ef244-bd15-4fb6-8980-3359db6cb323",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {