created experiment runner -- experimental version.

configs/config.yaml

@@ -0,0 +1,17 @@
+task: betti_numbers
+model:
+  model_name: GCN
+  hidden_channels: 64
+  num_hidden_layers: 3
+  num_node_features: 8
+  out_channels: 3
+trainer:
+  accelerator: cpu
+  max_epochs: 1
+  log_every_n_steps: 1
+litmodel:
+  learning_rate: 0.01
+data:
+  transforms: random_node_features
+  use_stratified: false
+  seed: 1234

configs/gcn_betti_numbers_config.yaml

@@ -0,0 +1,18 @@
+task: betti_numbers
+model:
+  model_name: GCN
+  hidden_channels: 64
+  num_hidden_layers: 3
+  num_node_features: 9
+  out_channels: 3
+
+trainer: 
+  accelerator: auto
+  max_epochs: 100
+  log_every_n_steps: 1
+
+litmodel:
+  learning_rate: .01
+
+data: 
+  use_stratified: False

configs/gcn_name_config.yaml

@@ -0,0 +1,18 @@
+task: name
+model:
+  model_name: GCN
+  hidden_channels: 64
+  num_hidden_layers: 3
+  num_node_features: 1
+  out_channels: 5
+
+trainer: 
+  accelerator: auto
+  max_epochs: 100
+  log_every_n_steps: 1
+
+litmodel:
+  learning_rate: .01
+
+data: 
+  use_stratified: True

configs/gcn_orientability_config.yaml

@@ -0,0 +1,18 @@
+task: orientability
+model:
+  model_name: GCN
+  hidden_channels: 64
+  num_hidden_layers: 3
+  num_node_features: 1
+  out_channels: 1
+
+trainer: 
+  accelerator: auto
+  max_epochs: 100
+  log_every_n_steps: 1
+
+litmodel:
+  learning_rate: .01
+
+data: 
+  use_stratified: True

configs/gcn_orientability_random_config.yaml

@@ -0,0 +1,20 @@
+task: orientability
+model:
+  model_name: GCN
+  hidden_channels: 64
+  num_hidden_layers: 3
+  num_node_features: 8
+  out_channels: 1
+
+trainer: 
+  accelerator: cpu
+  max_epochs: 100
+  log_every_n_steps: 1
+
+litmodel:
+  learning_rate: .01
+
+data: 
+  transforms: random_node_features
+  use_stratified: False
+  seed: 1234

experiment_betti_gnn.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 55,
+   "execution_count": 1,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2024-04-25T17:47:32.798261Z",
@@ -20,6 +20,7 @@
     "    TriangulationToFaceTransform,\n",
     "    OrientableToClassTransform,\n",
     "    DegreeTransform,\n",
+    "    degree_transform\n",
     ")\n",
     "from validation.validate_homology import validate_betti_numbers\n",
     "\n",
@@ -31,6 +32,57 @@
    "metadata": {},
    "source": []
   },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "tensor([[0., 0., 0., 1., 0., 0., 0., 0., 0.],\n",
+      "        [0., 0., 0., 1., 0., 0., 0., 0., 0.],\n",
+      "        [0., 0., 0., 1., 0., 0., 0., 0., 0.],\n",
+      "        [0., 0., 0., 1., 0., 0., 0., 0., 0.],\n",
+      "        [1., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
+      "        [1., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
+      "        [1., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
+      "        [1., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
+      "        [1., 0., 0., 0., 0., 0., 0., 0., 0.]])\n"
+     ]
+    }
+   ],
+   "source": [
+    "tr = transforms.Compose(\n",
+    "    [\n",
+    "        TriangulationToFaceTransform(),\n",
+    "        FaceToEdge(remove_faces=False),\n",
+    "        DegreeTransform(),\n",
+    "        OrientableToClassTransform(),\n",
+    "        OneHotDegree(max_degree=8, cat=False),\n",
+    "    ]\n",
+    ")\n",
+    "\n",
+    "class TriangulationToFaceTransform:\n",
+    "    def __call__(self, data):\n",
+    "        data.face = torch.tensor(data.triangulation).T - 1\n",
+    "        data.num_nodes = 9\n",
+    "        data.triangulation = None\n",
+    "        return data\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "dataset = SimplicialDataset(root=\"./data\",transform=tr)\n",
+    "dl = DataLoader(dataset, batch_size=1)  \n",
+    "\n",
+    "for batch in dl:\n",
+    "    print(batch.x)\n",
+    "    break\n"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 56,
@@ -63,17 +115,7 @@
     }
    ],
    "source": [
-    "tr = transforms.Compose(\n",
-    "    [\n",
-    "        TriangulationToFaceTransform(),\n",
-    "        FaceToEdge(remove_faces=False),\n",
-    "        DegreeTransform(),\n",
-    "        OrientableToClassTransform(),\n",
-    "        OneHotDegree(max_degree=8, cat=False),\n",
-    "    ]\n",
-    ")\n",
     "\n",
-    "dataset = SimplicialDataset(root=\"./data\", transform=tr)\n",
     "\n",
     "\n",
     "print()\n",

experiments/betti_numbers/graphs/GATSimplex2Vec.py

@@ -13,7 +13,7 @@
     DegreeTransform,
     Simplex2VecTransform,
 )
-from models.graphs.GAT import GATNetwork
+from models.GAT import GATNetwork
 
 
 class GATSimplexToVecModule(GraphCommonModuleBettiNumbers):

experiments/betti_numbers/graphs/GCN.py

@@ -8,7 +8,7 @@
 )
 from mantra.simplicial import SimplicialDataset
 from mantra.transforms import TriangulationToFaceTransform, DegreeTransform
-from models.graphs.GCN import GCNetwork
+from models.GCN import GCNetwork
 
 
 class GCNModule(GraphCommonModuleBettiNumbers):

experiments/betti_numbers/graphs/TAG.py

@@ -8,7 +8,7 @@
 )
 from mantra.simplicial import SimplicialDataset
 from mantra.transforms import TriangulationToFaceTransform, DegreeTransform
-from models.graphs.TAG import TAGNetwork
+from models.TAG import TAGNetwork
 
 
 class TAGModule(GraphCommonModuleBettiNumbers):

experiments/betti_numbers/graphs/TransformerConv.py

@@ -8,7 +8,7 @@
 )
 from mantra.simplicial import SimplicialDataset
 from mantra.transforms import TriangulationToFaceTransform, DegreeTransform
-from models.graphs.TransformerConv import TransformerConvNetwork
+from models.TransformerConv import TransformerConvNetwork
 
 
 class TransformerConvModule(GraphCommonModuleBettiNumbers):

experiments/betti_numbers/others/MLPConstantShape.py

@@ -9,7 +9,7 @@
 )
 from mantra.simplicial import SimplicialDataset
 from mantra.transforms import DegreeTransform, TriangulationToFaceTransform
-from models.others.MLPConstantShape import MLPConstantShape
+from models.MLPConstantShape import MLPConstantShape
 
 
 class MLPModule(BaseBettiNumbersModule):

experiments/name/graphs/GATSimplex2Vec.py

@@ -14,7 +14,7 @@
     Simplex2VecTransform,
     NameToClassTransform,
 )
-from models.graphs.GAT import GATNetwork
+from models.GAT import GATNetwork
 
 
 class GATSimplexToVecModule(GraphCommonModuleName):

experiments/name/graphs/GCN.py

@@ -12,7 +12,7 @@
     DegreeTransform,
     TriangulationToFaceTransform,
 )
-from models.graphs.GCN import GCNetwork
+from models.GCN import GCNetwork
 
 
 class GCNModule(GraphCommonModuleName):

experiments/name/graphs/TAG.py

@@ -12,7 +12,7 @@
     DegreeTransform,
     NameToClassTransform,
 )
-from models.graphs.TAG import TAGNetwork
+from models.TAG import TAGNetwork
 
 
 class TAGModule(GraphCommonModuleName):

experiments/name/graphs/TransformerConv.py

@@ -12,7 +12,7 @@
     DegreeTransform,
     NameToClassTransform,
 )
-from models.graphs.TransformerConv import TransformerConvNetwork
+from models.TransformerConv import TransformerConvNetwork
 
 
 class TransformerConvModule(GraphCommonModuleName):

experiments/name/others/MLPConstantShape.py

@@ -13,7 +13,7 @@
     TriangulationToFaceTransform,
     NameToClassTransform,
 )
-from models.others.MLPConstantShape import MLPConstantShape
+from models.MLPConstantShape import MLPConstantShape
 
 
 class MLPModule(BaseClassificationModule):

experiments/orientability/graphs/GATSimplex2Vec.py

@@ -14,7 +14,7 @@
     OrientableToClassTransform,
     Simplex2VecTransform,
 )
-from models.graphs.GAT import GATNetwork
+from models.GAT import GATNetwork
 
 
 class GATSimplexToVecModule(GraphCommonModuleOrientability):

experiments/orientability/graphs/GCN.py

@@ -12,7 +12,7 @@
     DegreeTransform,
     OrientableToClassTransform,
 )
-from models.graphs.GCN import GCNetwork
+from models.GCN import GCNetwork
 
 
 class GCNModule(GraphCommonModuleOrientability):

experiments/orientability/graphs/TAG.py

@@ -12,7 +12,7 @@
     DegreeTransform,
     OrientableToClassTransform,
 )
-from models.graphs.TAG import TAGNetwork
+from models.TAG import TAGNetwork
 
 
 class TAGModule(GraphCommonModuleOrientability):

experiments/orientability/graphs/TransformerConv.py

@@ -12,7 +12,7 @@
     DegreeTransform,
     OrientableToClassTransform,
 )
-from models.graphs.TransformerConv import TransformerConvNetwork
+from models.TransformerConv import TransformerConvNetwork
 
 
 class TransformerConvModule(GraphCommonModuleOrientability):

experiments/orientability/others/MLPConstantShape.py

@@ -13,7 +13,7 @@
     TriangulationToFaceTransform,
     OrientableToClassTransform,
 )
-from models.others.MLPConstantShape import MLPConstantShape
+from models.MLPConstantShape import MLPConstantShape
 
 
 class MLPModule(BaseClassificationModule):