[HWORKS-413] Add docs for model evaluation metrics (logicalclocks#176)

aversey · Feb 17, 2023 · 77b49f9 · 77b49f9
1 parent b9ac658
commit 77b49f9
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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`.