[
  {
    "metadataOutputVersion" : "3.0",
    "shortDescription" : "v9.1.2.thagomizer_32",
    "outputSchema" : [
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2)",
        "shortDescription" : "",
        "shape" : "[2]",
        "name" : "actionId",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1)",
        "shortDescription" : "",
        "shape" : "[1]",
        "name" : "actionCandidates",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2)",
        "shortDescription" : "",
        "shape" : "[2]",
        "name" : "shadowActionId",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1)",
        "shortDescription" : "",
        "shape" : "[1]",
        "name" : "shadowActionCandidates",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32)",
        "shortDescription" : "",
        "shape" : "[]",
        "name" : "rankings",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2)",
        "shortDescription" : "",
        "shape" : "[2]",
        "name" : "diagnostic",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 2)",
        "shortDescription" : "",
        "shape" : "[2, 2]",
        "name" : "anonymizedHistory",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1)",
        "shortDescription" : "",
        "shape" : "[1]",
        "name" : "forcedPrompt",
        "type" : "MultiArray"
      }
    ],
    "version" : "v9.1.2",
    "modelParameters" : [

    ],
    "specificationVersion" : 6,
    "storagePrecision" : "Float32",
    "mlProgramOperationTypeHistogram" : {
      "Greater" : 20,
      "Cast" : 43,
      "ReduceL2Norm" : 2,
      "Argsort" : 2,
      "SliceByIndex" : 30,
      "Log" : 4,
      "Concat" : 9,
      "Identity" : 10,
      "ScatterNd" : 6,
      "ReduceMax" : 4,
      "Inverse" : 3,
      "Fill" : 1,
      "ReduceMin" : 1,
      "Shape" : 9,
      "Linear" : 1,
      "Gather" : 1,
      "NotEqual" : 4,
      "Squeeze" : 2,
      "Equal" : 4,
      "RealDiv" : 10,
      "LogicalAnd" : 4,
      "Less" : 13,
      "Range1d" : 2,
      "NonZero" : 6,
      "Exp" : 9,
      "Mul" : 90,
      "Abs" : 5,
      "Scatter" : 2,
      "ExpandDims" : 28,
      "ReduceSum" : 15,
      "Pow" : 6,
      "GatherAlongAxis" : 2,
      "Sub" : 31,
      "ReduceProd" : 2,
      "Reshape" : 6,
      "Transpose" : 12,
      "Add" : 25,
      "Cos" : 2,
      "Tile" : 6
    },
    "computePrecision" : "Mixed (Float32, Int32)",
    "stateSchema" : [

    ],
    "isUpdatable" : "0",
    "availability" : {
      "macOS" : "12.0",
      "tvOS" : "15.0",
      "visionOS" : "1.0",
      "watchOS" : "8.0",
      "iOS" : "15.0",
      "macCatalyst" : "15.0"
    },
    "modelType" : {
      "name" : "MLModelType_mlProgram"
    },
    "inputSchema" : [
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "9 × 1...10000000 × 1...15",
        "shapeRange" : "[[9, 9], [1, 10000000], [1, 15]]",
        "formattedType" : "MultiArray (Float32 9 × 1 × 1)",
        "type" : "MultiArray",
        "shape" : "[9, 1, 1]",
        "name" : "tuples",
        "shortDescription" : ""
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 4)",
        "shortDescription" : "",
        "shape" : "[4]",
        "name" : "deviceContext",
        "type" : "MultiArray"
      },
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "2...10000 × 8",
        "shapeRange" : "[[2, 10000], [8, 8]]",
        "formattedType" : "MultiArray (Float32 2 × 8)",
        "type" : "MultiArray",
        "shape" : "[2, 8]",
        "name" : "candidateInteractions",
        "shortDescription" : ""
      },
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "1...1000 × 1...1000",
        "shapeRange" : "[[1, 1000], [1, 1000]]",
        "formattedType" : "MultiArray (Float32 1 × 1)",
        "type" : "MultiArray",
        "shape" : "[1, 1]",
        "name" : "tupleInteractions_candidates",
        "shortDescription" : ""
      },
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "1...1000 × 1...1000",
        "shapeRange" : "[[1, 1000], [1, 1000]]",
        "formattedType" : "MultiArray (Float32 1 × 1)",
        "type" : "MultiArray",
        "shape" : "[1, 1]",
        "name" : "tupleInteractions_alignment",
        "shortDescription" : ""
      },
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "1...10 × 3",
        "shapeRange" : "[[1, 10], [3, 3]]",
        "formattedType" : "MultiArray (Float32 1 × 3)",
        "type" : "MultiArray",
        "shape" : "[1, 3]",
        "name" : "similarityScores",
        "shortDescription" : ""
      },
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "1...10000000 × 2",
        "shapeRange" : "[[1, 10000000], [2, 2]]",
        "formattedType" : "MultiArray (Float32 1 × 2)",
        "type" : "MultiArray",
        "shape" : "[1, 2]",
        "name" : "riskLevel",
        "shortDescription" : ""
      },
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "1...10000000 × 1...15",
        "shapeRange" : "[[1, 10000000], [1, 15]]",
        "formattedType" : "MultiArray (Float32 1 × 1)",
        "type" : "MultiArray",
        "shape" : "[1, 1]",
        "name" : "candidate_risk",
        "shortDescription" : ""
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 4)",
        "shortDescription" : "",
        "shape" : "[4]",
        "name" : "forcedPromptRate",
        "type" : "MultiArray"
      },
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "1...15",
        "shapeRange" : "[[1, 15]]",
        "formattedType" : "MultiArray (Float32 1)",
        "type" : "MultiArray",
        "shape" : "[1]",
        "name" : "parameterName",
        "shortDescription" : ""
      },
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "1...15",
        "shapeRange" : "[[1, 15]]",
        "formattedType" : "MultiArray (Float32 1)",
        "type" : "MultiArray",
        "shape" : "[1]",
        "name" : "alreadyPrompted",
        "shortDescription" : ""
      },
      {
        "dataType" : "Float32",
        "hasShapeFlexibility" : "1",
        "isOptional" : "0",
        "shapeFlexibility" : "1...15",
        "shapeRange" : "[[1, 15]]",
        "formattedType" : "MultiArray (Float32 1)",
        "type" : "MultiArray",
        "shape" : "[1]",
        "name" : "isResolved",
        "shortDescription" : ""
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1)",
        "shortDescription" : "",
        "shape" : "[1]",
        "name" : "component",
        "type" : "MultiArray"
      }
    ],
    "userDefinedMetadata" : {
      "com.github.apple.coremltools.version" : "8.2.2",
      "com.github.apple.coremltools.source_dialect" : "TorchScript",
      "com.github.apple.coremltools.source" : "torch==2.6.0",
      "tag" : "mldb-wrxg5d7ao8",
      "mldb_token" : "mldb-wrxg5d7ao8"
    },
    "generatedClassName" : "ASRRankingModel",
    "method" : "predict"
  }
]