From 77b49f9f8ecd8e7a0c803d50d71a0baf3b8ec9ba Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Javier=20de=20la=20R=C3=BAa=20Mart=C3=ADnez?= Date: Fri, 17 Feb 2023 14:39:17 +0100 Subject: [PATCH] [HWORKS-413] Add docs for model evaluation metrics (#176) --- python/hsml/python/signature.py | 1 + python/hsml/sklearn/signature.py | 1 + python/hsml/tensorflow/signature.py | 1 + python/hsml/torch/signature.py | 1 + 4 files changed, 4 insertions(+) diff --git a/python/hsml/python/signature.py b/python/hsml/python/signature.py index 80e6ff9ca..96a7d851f 100644 --- a/python/hsml/python/signature.py +++ b/python/hsml/python/signature.py @@ -45,6 +45,7 @@ def create_model( version: Optionally version of the model to create, defaults to `None` and will create the model with incremented version from the last version in the model registry. + metrics: Optionally a dictionary with model evaluation metrics (e.g., accuracy, MAE) description: Optionally a string describing the model, defaults to empty string `""`. input_example: Optionally an input example that represents a single input for the model, defaults to `None`. diff --git a/python/hsml/sklearn/signature.py b/python/hsml/sklearn/signature.py index 4af911f40..89fd268b9 100644 --- a/python/hsml/sklearn/signature.py +++ b/python/hsml/sklearn/signature.py @@ -45,6 +45,7 @@ def create_model( version: Optionally version of the model to create, defaults to `None` and will create the model with incremented version from the last version in the model registry. + metrics: Optionally a dictionary with model evaluation metrics (e.g., accuracy, MAE) description: Optionally a string describing the model, defaults to empty string `""`. input_example: Optionally an input example that represents a single input for the model, defaults to `None`. diff --git a/python/hsml/tensorflow/signature.py b/python/hsml/tensorflow/signature.py index 5ea2d9754..022e8e731 100644 --- a/python/hsml/tensorflow/signature.py +++ b/python/hsml/tensorflow/signature.py @@ -45,6 +45,7 @@ def create_model( version: Optionally version of the model to create, defaults to `None` and will create the model with incremented version from the last version in the model registry. + metrics: Optionally a dictionary with model evaluation metrics (e.g., accuracy, MAE) description: Optionally a string describing the model, defaults to empty string `""`. input_example: Optionally an input example that represents a single input for the model, defaults to `None`. diff --git a/python/hsml/torch/signature.py b/python/hsml/torch/signature.py index 9cff09d53..3aed923cd 100644 --- a/python/hsml/torch/signature.py +++ b/python/hsml/torch/signature.py @@ -45,6 +45,7 @@ def create_model( version: Optionally version of the model to create, defaults to `None` and will create the model with incremented version from the last version in the model registry. + metrics: Optionally a dictionary with model evaluation metrics (e.g., accuracy, MAE) description: Optionally a string describing the model, defaults to empty string `""`. input_example: Optionally an input example that represents a single input for the model, defaults to `None`.