chore(ingest): add example of training metric/hyper parameters (#8491)

datahub-project · Jul 25, 2023 · cc46729 · cc46729
1 parent bf9f380
commit cc46729
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/metadata-ingestion/examples/library/create_mlmodel.py b/metadata-ingestion/examples/library/create_mlmodel.py
@@ -31,6 +31,16 @@
         description="my feature",
         groups=model_group_urns,
         mlFeatures=feature_urns,
+        trainingMetrics=[
+            models.MLMetricClass(
+                name="accuracy", description="accuracy of the model", value="1.0"
+            )
+        ],
+        hyperParams=[
+            models.MLHyperParamClass(
+                name="hyper_1", description="hyper_1", value="0.102"
+            )
+        ],
     ),
 )