adding new metric loss_batch from merlin models (#727)

jperez999 · jperez999 · commit e334d75a1510 · 2022-11-21T08:31:59.000-05:00
diff --git a/tests/unit/examples/test_scaling_criteo_merlin_models.py b/tests/unit/examples/test_scaling_criteo_merlin_models.py
@@ -59,6 +59,7 @@ def test_func():
                 "precision",
                 "recall",
                 "regularization_loss",
+                "loss_batch",
             ]
         )
         assert os.path.isfile("/tmp/output/criteo/dlrm/saved_model.pb")

Original file line number	Diff line number	Diff line change
`@@ -59,6 +59,7 @@ def test_func():`
`59`	`59`	`"precision",`
`60`	`60`	`"recall",`
`61`	`61`	`"regularization_loss",`
	`62`	`+ "loss_batch",`
`62`	`63`	`]`
`63`	`64`	`)`
`64`	`65`	`assert os.path.isfile("/tmp/output/criteo/dlrm/saved_model.pb")`