config.json

{
  "model_type": "casting-solidification-gnn",
  "version": "v5.1",
  "code": "https://github.com/eugenmik/castsolid-gnn",
  "files": {
    "gnn": "gnn_v5.1.safetensors",
    "total_head": "gbm_total_v3.skops",
    "spread_head": "gbm_std_v2.skops",
    "ood_ranges": "ood_ranges.json"
  },
  "gnn": {
    "architecture": "MeshGraphNet-style encode-process-decode with FiLM conditioning",
    "node_features": 15,
    "edge_features": 5,
    "conditioning_features": 5,
    "hidden": 128,
    "message_passing_blocks": 12,
    "knn_neighbours": 16,
    "graph": "tetrahedral mesh edges union k-nearest neighbours"
  },
  "heads": {
    "total_head_features": 16,
    "spread_head_features": 17,
    "sklearn_version": "1.5.2"
  },
  "recomposition": {
    "method": "soft_anchor",
    "quantile": 0.995,
    "w_total": 1.0,
    "w_std": 4.0
  },
  "input": {
    "mesh": "gmsh 2.2 ASCII tetrahedral mesh in mm, cooling surface tagged 'Cooling'",
    "mesh_nodes_trained": [7500, 10000],
    "properties": ["k", "rho", "cp", "L_kJ_kg", "TL", "TS", "Tpour", "mold_effusivity"]
  },
  "output": "solidification time in seconds per mesh node"
}