File tree Expand file tree Collapse file tree 1 file changed +11
-0
lines changed
Expand file tree Collapse file tree 1 file changed +11
-0
lines changed Original file line number Diff line number Diff 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+
5061message SoftmaxCrossEntropyWithNegativeMining {
5162 required uint32 num_negative_samples = 1 ;
5263 required float margin = 2 [default = 0 ];
You can’t perform that action at this time.
0 commit comments