add ZILN loss for ltv prediction task

alibaba · Nov 12, 2024 · 60ca6d8 · 60ca6d8
1 parent 38dfe80
commit 60ca6d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/easy_rec/python/model/rank_model.py b/easy_rec/python/model/rank_model.py
@@ -296,7 +296,7 @@ def _build_metric_impl(self,
         LossType.PAIR_WISE_LOSS, LossType.BINARY_FOCAL_LOSS,
         LossType.PAIRWISE_FOCAL_LOSS, LossType.PAIRWISE_LOGISTIC_LOSS,
         LossType.JRC_LOSS, LossType.LISTWISE_DISTILL_LOSS,
-        LossType.LISTWISE_RANK_LOSS
+        LossType.LISTWISE_RANK_LOSS, LossType.ZILN_LOSS
     }
     metric_dict = {}
     if metric.WhichOneof('metric') == 'auc':