Skip to content

Commit bd230cb

Browse files
authored
add proto config for ziln loss (#547)
* upgrade zero inflated lognormal loss
1 parent aec31c4 commit bd230cb

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

easy_rec/python/protos/loss.proto

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,20 @@ message Loss {
4444
PairwiseHingeLoss pairwise_hinge_loss = 110;
4545
ListwiseRankLoss listwise_rank_loss = 111;
4646
ListwiseDistillLoss listwise_distill_loss = 112;
47+
ZILNLoss ziln_loss = 113;
4748
}
4849
};
4950

51+
message ZILNLoss {
52+
optional float mu_regularization = 1 [default = 0.01];
53+
optional float sigma_regularization = 2 [default = 0.01];
54+
optional float max_log_clip_value = 3 [default = 20.0];
55+
optional float max_sigma = 4 [default = 5.0];
56+
optional bool return_log_pred_value = 5 [default = false];
57+
optional float classification_weight = 6 [default = 1.0];
58+
optional float regression_weight = 7 [default = 1.0];
59+
}
60+
5061
message SoftmaxCrossEntropyWithNegativeMining {
5162
required uint32 num_negative_samples = 1;
5263
required float margin = 2 [default = 0];

0 commit comments

Comments
 (0)