Skip to content

Document available outputs in SKLearn runtime #997

Description

@adriangonz

The SKLearn runtime lets you choose which model method you want as part of the outputs field of the request payload. However, this is not documented at all at the moment. Depending on the model, the available outputs are:

  • predict
  • predict_proba
  • transform

E.g. to ask for both the predict and predict_proba outputs, you could do:

{
    "inputs": [
        {
          "name": "payload",
          "shape": [2, 3],
          "datatype": "FP32",
          "data": [1, 2, 3, 4]
        }
    ],
   "outputs": [
    { "name": "predict" },
    { "name": "predict_proba" }
  ]
}

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions