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`.