Update to use new eval results folder

adamkarvonen · Oct 2, 2024 · 69657d8 · 69657d8
1 parent fdc0afd
commit 69657d8
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/template.ipynb b/template.ipynb
@@ -106,7 +106,7 @@
     "        )\n",
     "\n",
     "for key in selected_saes_dict:\n",
-    "    print(key, selected_saes_dict[key])"
+    "    print(\"\\n\\n\", key, \"\\n\\n\",selected_saes_dict[key])"
    ]
   },
   {
@@ -141,7 +141,7 @@
     "        )\n",
     "\n",
     "for key in selected_saes_dict:\n",
-    "    print(key, selected_saes_dict[key])"
+    "    print(\"\\n\\n\", key, \"\\n\\n\",selected_saes_dict[key])"
    ]
   },
   {
@@ -177,7 +177,7 @@
     "        )\n",
     "\n",
     "for key in selected_saes_dict:\n",
-    "    print(key, selected_saes_dict[key])"
+    "    print(\"\\n\\n\", key, \"\\n\\n\",selected_saes_dict[key])"
    ]
   },
   {
@@ -320,15 +320,15 @@
    "outputs": [],
    "source": [
     "folder_path = \"sparse_probing/sparse_probing_results\"\n",
-    "filename = f\"example_results_{release_name}_eval_results.json\"\n",
+    "filename = \"example_gemma-2-2b_layer_19_eval_results.json\"\n",
     "filepath = os.path.join(folder_path, filename)\n",
     "\n",
     "with open(filepath, \"r\") as f:\n",
     "    custom_eval_results = json.load(f)\n",
     "\n",
     "print(custom_eval_results.keys())\n",
     "print(f'\\nCustom eval config:\\n{custom_eval_results[\"custom_eval_config\"]}')\n",
-    "print(f'\\nCustom eval results:\\n{custom_eval_results[\"custom_eval_results\"][example_sae_name]}')"
+    "print(f'\\nCustom eval results for {example_sae_name}:\\n{custom_eval_results[\"custom_eval_results\"][example_sae_name]}')"
    ]
   }
  ],
@@ -348,7 +348,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.13"
+   "version": "3.11.8"
   }
  },
  "nbformat": 4,