json.schema.api.data.createMlModel.json Maven / Gradle / Ivy

Go to download
{
  "$id": "https://open-metadata.org/schema/api/data/createMlModel.json",
  "$schema": "http://json-schema.org/draft-07/schema#",
  "title": "CreateMlModelRequest",
  "description": "Create Ml Model entity request",
  "type": "object",
  "javaType": "org.openmetadata.catalog.api.data.CreateMlModel",
  "javaInterfaces": ["org.openmetadata.core.entity.interfaces.CreateEntity"],

  "properties": {
    "name": {
      "description": "Name that identifies this ML model.",
      "$ref": "../../type/basic.json#/definitions/entityName"
    },
    "displayName": {
      "description": "Display Name that identifies this ML model. It could be title or label from the source services",
      "type": "string"
    },
    "description": {
      "description": "Description of the ML model instance. How it was trained and for what it is used.",
      "$ref": "../../type/basic.json#/definitions/markdown"
    },
    "algorithm": {
      "description": "Algorithm used to train the ML Model",
      "type": "string"
    },
    "mlFeatures": {
      "description": "Features used to train the ML Model.",
      "type": "array",
      "items": {
        "$ref": "../../entity/data/mlmodel.json#/definitions/mlFeature"
      },
      "default": null
    },
    "target": {
      "description": "For supervised ML Models, the value to estimate.",
      "$ref": "../../type/basic.json#/definitions/entityName"
    },
    "mlHyperParameters": {
      "description": "Hyper Parameters used to train the ML Model.",
      "type": "array",
      "items": {
        "$ref": "../../entity/data/mlmodel.json#/definitions/mlHyperParameter"
      },
      "default": null
    },
    "dashboard": {
      "description": "Performance Dashboard URL to track metric evolution",
      "$ref": "../../type/entityReference.json"
    },
    "mlStore": {
      "description": "Location containing the ML Model. It can be a storage layer and/or a container repository.",
      "$ref": "../../entity/data/mlmodel.json#/definitions/mlStore"
    },
    "server": {
      "description": "Endpoint that makes the ML Model available, e.g,. a REST API serving the data or computing predictions.",
      "$ref": "../../type/basic.json#/definitions/href"
    },
    "tags": {
      "description": "Tags for this ML Model",
      "type": "array",
      "items": {
        "$ref": "../../type/tagLabel.json"
      },
      "default": null
    },
    "owner": {
      "description": "Owner of this database",
      "$ref": "../../type/entityReference.json"
    },
    "service": {
      "description": "Link to the pipeline service where this pipeline is hosted in",
      "$ref": "../../type/entityReference.json"
    },
    "extension": {
      "description": "Entity extension data with custom attributes added to the entity.",
      "$ref": "../../type/basic.json#/definitions/entityExtension"
    }
  },
  "required": ["name", "algorithm", "service"],
  "additionalProperties": false
}