{ "$schema": "http://json-schema.org/schema#", "$schemaVersion": "0.0.2", "modelTags": "", "$id": "https://smart-data-models.github.io/dataModel.MLModel/MLModel/schema.json", "title": "Smart Data Models - Machine Learning Model", "description": "Data model for compilation of the elements of a machine learning model.", "type": "object", "allOf": [ { "$ref": "https://smart-data-models.github.io/data-models/common-schema.json#/definitions/GSMA-Commons" }, { "properties": { "type": { "type": "string", "description": "Property. NGSI entity type. It has to be MLModel", "enum": [ "MLModel" ] }, "typeOfAlgorithm": { "type": "string", "description": "Property. Model:'https://schema.org/Text'. Enumeration" }, "algorithm": { "type": "string", "description": "Property. The algorithm used by the underlying Machine Learning model (e.g. linear regression, k-means, SVM, MLP,...)" }, "mlFramework": { "type": "string", "description": "Property. The Machine Learning framework that has been used to prepare the model (e.g., scikit-learn, H2O, Spark MLib, etc)" }, "version": { "type": "string", "description": "Property. Version of the model." }, "acceptableDataSources": { "type": "array", "description": "Property. Valid type of input data sources for running the Machine Learning Model", "items": { "type": "string" } }, "outputDataTypes": { "type": "array", "description": "Property. Type of output data produced by the Machine Learning Model", "items": { "type": "string" } }, "dockerImage": { "type": "string", "description": "Property. Docker image containing the Machine Learning model" }, "refMLProcessing": { "type": "array", "description": "Property", "items": { "type": "string", "format": "uri" } }, "inputAttributes": { "type": "array", "description": "Property. Comma-separated list of attributes names (that should have a given type by definition).", "items": { "type": "string" } }, "outputAttributes": { "type": "array", "description": "Property. Comma-separated list of attributes names used to publish the results.", "items": { "type": "string" } } } } ], "required": [ "id", "type" ] }