geometric_classifier v8 | epoch 79 | acc=0.9022 | 1,852,870 params

geometric_classifier/config.json

@@ -0,0 +1,260 @@
+{
+  "model_type": "GeometricShapeClassifier",
+  "version": "v8",
+  "grid_size": 5,
+  "num_classes": 38,
+  "class_names": [
+    "point",
+    "line_x",
+    "line_y",
+    "line_z",
+    "line_diag",
+    "cross",
+    "l_shape",
+    "collinear",
+    "triangle_xy",
+    "triangle_xz",
+    "triangle_3d",
+    "square_xy",
+    "square_xz",
+    "rectangle",
+    "coplanar",
+    "plane",
+    "tetrahedron",
+    "pyramid",
+    "pentachoron",
+    "cube",
+    "cuboid",
+    "triangular_prism",
+    "octahedron",
+    "arc",
+    "helix",
+    "circle",
+    "ellipse",
+    "disc",
+    "sphere",
+    "hemisphere",
+    "cylinder",
+    "cone",
+    "capsule",
+    "torus",
+    "shell",
+    "tube",
+    "bowl",
+    "saddle"
+  ],
+  "curvature_types": [
+    "none",
+    "convex",
+    "concave",
+    "cylindrical",
+    "conical",
+    "toroidal",
+    "hyperbolic",
+    "helical"
+  ],
+  "embed_dim": 128,
+  "n_tracers": 5,
+  "capacity_dims": [
+    64,
+    64,
+    64,
+    64
+  ],
+  "curvature_embed_dim": 128,
+  "arbiter_latent_dim": 128,
+  "arbiter_flow_steps": 4,
+  "total_params": 1852870,
+  "shape_catalog": {
+    "point": {
+      "dim": 0,
+      "curved": false,
+      "curvature": "none"
+    },
+    "line_x": {
+      "dim": 1,
+      "curved": false,
+      "curvature": "none"
+    },
+    "line_y": {
+      "dim": 1,
+      "curved": false,
+      "curvature": "none"
+    },
+    "line_z": {
+      "dim": 1,
+      "curved": false,
+      "curvature": "none"
+    },
+    "line_diag": {
+      "dim": 1,
+      "curved": false,
+      "curvature": "none"
+    },
+    "cross": {
+      "dim": 1,
+      "curved": false,
+      "curvature": "none"
+    },
+    "l_shape": {
+      "dim": 1,
+      "curved": false,
+      "curvature": "none"
+    },
+    "collinear": {
+      "dim": 1,
+      "curved": false,
+      "curvature": "none"
+    },
+    "triangle_xy": {
+      "dim": 2,
+      "curved": false,
+      "curvature": "none"
+    },
+    "triangle_xz": {
+      "dim": 2,
+      "curved": false,
+      "curvature": "none"
+    },
+    "triangle_3d": {
+      "dim": 2,
+      "curved": false,
+      "curvature": "none"
+    },
+    "square_xy": {
+      "dim": 2,
+      "curved": false,
+      "curvature": "none"
+    },
+    "square_xz": {
+      "dim": 2,
+      "curved": false,
+      "curvature": "none"
+    },
+    "rectangle": {
+      "dim": 2,
+      "curved": false,
+      "curvature": "none"
+    },
+    "coplanar": {
+      "dim": 2,
+      "curved": false,
+      "curvature": "none"
+    },
+    "plane": {
+      "dim": 2,
+      "curved": false,
+      "curvature": "none"
+    },
+    "tetrahedron": {
+      "dim": 3,
+      "curved": false,
+      "curvature": "none"
+    },
+    "pyramid": {
+      "dim": 3,
+      "curved": false,
+      "curvature": "none"
+    },
+    "pentachoron": {
+      "dim": 3,
+      "curved": false,
+      "curvature": "none"
+    },
+    "cube": {
+      "dim": 3,
+      "curved": false,
+      "curvature": "none"
+    },
+    "cuboid": {
+      "dim": 3,
+      "curved": false,
+      "curvature": "none"
+    },
+    "triangular_prism": {
+      "dim": 3,
+      "curved": false,
+      "curvature": "none"
+    },
+    "octahedron": {
+      "dim": 3,
+      "curved": false,
+      "curvature": "none"
+    },
+    "arc": {
+      "dim": 1,
+      "curved": true,
+      "curvature": "convex"
+    },
+    "helix": {
+      "dim": 1,
+      "curved": true,
+      "curvature": "helical"
+    },
+    "circle": {
+      "dim": 2,
+      "curved": true,
+      "curvature": "convex"
+    },
+    "ellipse": {
+      "dim": 2,
+      "curved": true,
+      "curvature": "convex"
+    },
+    "disc": {
+      "dim": 2,
+      "curved": true,
+      "curvature": "convex"
+    },
+    "sphere": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "convex"
+    },
+    "hemisphere": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "convex"
+    },
+    "cylinder": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "cylindrical"
+    },
+    "cone": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "conical"
+    },
+    "capsule": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "convex"
+    },
+    "torus": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "toroidal"
+    },
+    "shell": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "convex"
+    },
+    "tube": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "cylindrical"
+    },
+    "bowl": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "concave"
+    },
+    "saddle": {
+      "dim": 3,
+      "curved": true,
+      "curvature": "hyperbolic"
+    }
+  }
+}

geometric_classifier/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7b9d0a0144d9c5746b93fb734ff73df5991cba83e4d25c5cf2258ff17b22c3b1
+size 7433664

geometric_classifier/training_config.json

@@ -0,0 +1,10 @@
+{
+  "checkpoint_epoch": 79,
+  "best_val_accuracy": 0.90219,
+  "learned_capacities": [
+    0.004316786769777536,
+    0.01499153021723032,
+    0.07568630576133728,
+    16.490644454956055
+  ]
+}