additional plots

1 files changed, 22 insertions, 14 deletions

diff --git a/notebooks/cc/references.ipynb b/notebooks/cc/references.ipynb
index e474d816..4b027a55 100644
--- a/notebooks/cc/references.ipynb
+++ b/notebooks/cc/references.ipynb
@@ -30,6 +30,7 @@
     "from collections.abc import Iterable\n",
     "from pathlib import Path\n",
     "\n",
+    "import matplotlib\n",
     "import matplotlib.pyplot as plt\n",
     "import networkx as nx\n",
     "import networkx.algorithms.community as nx_comm\n",
@@ -46,7 +47,14 @@
     "\n",
     "%matplotlib inline\n",
     "\n",
+    "# LaTeX plotting\n",
     "# matplotlib.use(\"pgf\")\n",
+    "# plt.rcParams[\"pgf.texsystem\"] = \"pdflatex\"\n",
+    "# plt.rcParams[\"font.family\"] = \"serif\"\n",
+    "# plt.rcParams[\"text.usetex\"] = True\n",
+    "# plt.rcParams[\"pgf.rcfonts\"] = False\n",
+    "\n",
+    "\n",
     "sns.set_theme(style=\"white\")\n",
     "plt.rcParams[\"axes.linewidth\"] = 0.5\n",
     "plt.rcParams[\"legend.fontsize\"] = 6.5\n",
@@ -60,10 +68,7 @@
     "plt.rcParams[\"xtick.major.size\"] = 5\n",
     "plt.rcParams[\"xtick.major.width\"] = 0.5\n",
     "plt.rcParams[\"xtick.major.pad\"] = 0\n",
-    "# plt.rcParams[\"pgf.texsystem\"] = \"pdflatex\"\n",
-    "# plt.rcParams[\"font.family\"] = \"serif\"\n",
-    "# plt.rcParams[\"text.usetex\"] = True\n",
-    "# plt.rcParams[\"pgf.rcfonts\"] = False\n",
+    "\n",
     "plt.rcParams[\"axes.titlesize\"] = 8\n",
     "plt.rcParams[\"legend.handletextpad\"] = 0.3\n",
     "plt.rcParams[\"lines.markersize\"] = 4\n",
@@ -76,9 +81,12 @@
     "\n",
     "# plt.rcParams['figure.figsize'] = (10, 6)\n",
     "\n",
+    "REPO_ROOT = Path().resolve()\n",
     "RESULTS_DIR = Path(\"./results/references\")\n",
     "RESULTS_DIR.mkdir(exist_ok=True, parents=True)\n",
-    "SMARTCARD_CATEGORY = \"ICs, Smart Cards and Smart Card-Related Devices and Systems\""
+    "SMARTCARD_CATEGORY = \"ICs, Smart Cards and Smart Card-Related Devices and Systems\"\n",
+    "DATASET_PATH = REPO_ROOT / \"dataset/cc_november_23/dataset.json\"\n",
+    "PREDICTIONS_PATH = REPO_ROOT / \"dataset/reference_prediction/predictions.csv\""
    ]
   },
   {
@@ -193,9 +201,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "dset = CCDataset.from_json(\"/var/tmp/xjanovsk/certs/sec-certs/dataset/cc_november_23/dataset.json\")\n",
+    "dset = CCDataset.from_json(DATASET_PATH)\n",
     "cc_df = preprocess_cc_df(dset.to_pandas())\n",
-    "refs_df = preprocess_refs_df(\"/var/tmp/xjanovsk/certs/sec-certs/dataset/reference_prediction/predictions.csv\", cc_df)\n",
+    "refs_df = preprocess_refs_df(PREDICTIONS_PATH, cc_df)\n",
     "unique_labels = refs_df.reference_label.unique().tolist()\n",
     "\n",
     "# Load labeled reference graph as networkx directed graph\n",
@@ -214,13 +222,6 @@
    ]
   },
   {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Common processing functions"
-   ]
-  },
-  {
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
@@ -237,6 +238,13 @@
    ]
   },
   {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Common processing functions"
+   ]
+  },
+  {
    "attachments": {},
    "cell_type": "markdown",
    "metadata": {},