geometric_classifier/config.json

{
  "model_type": "GeometricShapeClassifier",
  "version": "v9.1",
  "changes": "enhanced arbiter: 256-dim latent, 8 flow steps, logit conditioning, adaptive blend, cosine prototypes",
  "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": 64,
  "n_tracers": 5,
  "arbiter_latent_dim": 256,
  "arbiter_flow_steps": 8,
  "total_params": 3486678,
  "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"
    }
  }
}