bpr_loss
bpr_loss
(input, label, name=None)[源代码]
该OP属于pairwise类型的损失函数。损失值由下式计算而得:
- 其中:
- : 在时间步i的正例和负例的总和。
:在时间步i的正例下标。
- :激活函数。
:输出值,一个形为[T x 1]的2-D Tensor。
更多细节请参考 Session Based Recommendations with Recurrent Neural Networks
- input (Variable) - 形为[T x D] , Tensor类型时T为batch大小,LoDTensor类型时T为mini-batch的总时间步。D 为正例加负例的个数。该输入为logits而非概率。数据类型是float32或float64。
- label (Variable) - 形为[T x 1],表示input中正例的下标,数据类型为int64。。
- name (None|str) – 该参数供开发人员打印调试信息时使用,具体用法请参见 ,默认值为None。
返回类型:Variable