Skip to content
This repository has been archived by the owner on Oct 2, 2024. It is now read-only.

hyperparameter definition #14

Closed
Tracked by #7
fmigneault opened this issue Mar 26, 2024 · 0 comments
Closed
Tracked by #7

hyperparameter definition #14

fmigneault opened this issue Mar 26, 2024 · 0 comments
Assignees

Comments

@fmigneault
Copy link
Collaborator

fmigneault commented Mar 26, 2024

When the model being described needs to provide hyperparameters (for example, the number of estimators used in a scikit-learn model, or a learning rate for an optimizer), there is no clear way currently to define them.

One proposal would be to have a definition as the following under a STAC Item:

{
  "properties": {
    "mlm:hyperparameters": [
      {
        "name": "n_estimators",
        "value": 2
      }
    ]
  }
}

The mlm:hyperparameters could also be allowed under a specific Asset, in case the user wants to be more specific about which Asset they apply to.

{
  "assets": {
    "source": {
      "title": "Model training pipeline",
      "href": "<URI-to-python-training-script>",
      "type": "text/x-python",
      "mlm:hyperparameters": [
        {
          "name": "n_estimators",
          "value": 2
        }
      ],
      "roles": [
        "metadata",
        "code",
        "model"
      ]
    } 
  }
}
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant