Energy + inelasticity regression

pyproject.toml

@@ -23,6 +23,7 @@ dependencies = [
     "torch-geometric (>=2.6.1,<3.0.0)",
     "awkward-pandas (>=2023.8.0,<2024.0.0)",
     "seaborn (>=0.13.2,<0.14.0)",
+    "particle (>=0.26.1,<0.27.0)",
 ]
 
 [tool.poetry]

src/analysis/cnn.py → src/analysis/cnn/cnn.py

@@ -7,7 +7,7 @@
 from torch.utils.data import DataLoader
 
 from analysis.utils.datasets import CNN3DDataset, CNNProjectionDataset
-from analysis.utils.networks import ClassifierProjectionCNN
+from analysis.utils.models import ClassifierProjectionCNN
 from analysis.utils.torch_data_utils import evaluate_model
 from analysis.utils.torch_utils import get_device, show_model, train
 from analysis.utils.utils import get_figures_path, get_torch_path, get_weights_path

src/analysis/evaluate_cnn.ipynb → src/analysis/cnn/evaluate_cnn.ipynb

@@ -96,15 +96,19 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "26509ddd",
    "metadata": {},
    "outputs": [
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Using device: cpu\n"
+     "ename": "NameError",
+     "evalue": "name 'get_torch_path' is not defined",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[31m---------------------------------------------------------------------------\u001b[39m",
+      "\u001b[31mNameError\u001b[39m                                 Traceback (most recent call last)",
+      "\u001b[36mCell\u001b[39m\u001b[36m \u001b[39m\u001b[32mIn[1]\u001b[39m\u001b[32m, line 5\u001b[39m\n\u001b[32m      2\u001b[39m \u001b[38;5;66;03m# geometry_label = \"cnn_projection_200um_bins_10000_10003\"\u001b[39;00m\n\u001b[32m      3\u001b[39m \u001b[38;5;66;03m# geometry_label = \"theta_100mrad_ratio_10\"\u001b[39;00m\n\u001b[32m      4\u001b[39m geometry_label = \u001b[33m\"\u001b[39m\u001b[33m./\u001b[39m\u001b[33m\"\u001b[39m\n\u001b[32m----> \u001b[39m\u001b[32m5\u001b[39m torch_path = \u001b[43mget_torch_path\u001b[49m() / geometry_label / \u001b[33m\"\u001b[39m\u001b[33mcnn.pt\u001b[39m\u001b[33m\"\u001b[39m\n\u001b[32m      6\u001b[39m weights_path = get_weights_path() / geometry_label\n\u001b[32m      7\u001b[39m figures_path = get_figures_path() / geometry_label\n",
+      "\u001b[31mNameError\u001b[39m: name 'get_torch_path' is not defined"
      ]
     }
    ],
@@ -135,7 +139,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 165,
+   "execution_count": null,
    "id": "a7bafe29",
    "metadata": {},
    "outputs": [
@@ -155,7 +159,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 166,
+   "execution_count": null,
    "id": "c5220938",
    "metadata": {},
    "outputs": [
@@ -180,7 +184,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 167,
+   "execution_count": null,
    "id": "65279494",
    "metadata": {},
    "outputs": [
@@ -201,7 +205,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 168,
+   "execution_count": null,
    "id": "bcc4a46a",
    "metadata": {},
    "outputs": [
@@ -224,7 +228,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 169,
+   "execution_count": null,
    "id": "b828959e",
    "metadata": {},
    "outputs": [
@@ -247,7 +251,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 170,
+   "execution_count": null,
    "id": "a3ff6371",
    "metadata": {},
    "outputs": [
@@ -273,7 +277,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 171,
+   "execution_count": null,
    "id": "0fe4eed0",
    "metadata": {},
    "outputs": [
@@ -343,7 +347,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 175,
+   "execution_count": null,
    "id": "3b474d45",
    "metadata": {},
    "outputs": [],
@@ -365,7 +369,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 190,
+   "execution_count": null,
    "id": "d92a7229",
    "metadata": {},
    "outputs": [],
@@ -379,7 +383,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 210,
+   "execution_count": null,
    "id": "62886ec4",
    "metadata": {},
    "outputs": [],
@@ -391,7 +395,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 202,
+   "execution_count": null,
    "id": "2ffd83fc",
    "metadata": {},
    "outputs": [],
@@ -409,7 +413,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 215,
+   "execution_count": null,
    "id": "ccbfcc35",
    "metadata": {},
    "outputs": [],
@@ -423,7 +427,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 219,
+   "execution_count": null,
    "id": "4a5397e8",
    "metadata": {},
    "outputs": [
@@ -446,7 +450,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 230,
+   "execution_count": null,
    "id": "87291b5f",
    "metadata": {},
    "outputs": [
@@ -482,7 +486,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 231,
+   "execution_count": null,
    "id": "b051a4c8",
    "metadata": {},
    "outputs": [