|
|
|
@ -5830,24 +5830,25 @@ def rank_loss(label, left, right, name=None):
|
|
|
|
|
|
|
|
|
|
def margin_rank_loss(label, left, right, margin=0.1, name=None):
|
|
|
|
|
"""
|
|
|
|
|
Margin Rank loss layer for rank problem, which comparing left value and right value be passed in.
|
|
|
|
|
The rank loss can be defined as below equation:
|
|
|
|
|
Margin Ranking Loss Layer for ranking problem,
|
|
|
|
|
which compare left score and right score passed in.
|
|
|
|
|
The ranking loss can be defined as following equation:
|
|
|
|
|
|
|
|
|
|
.. math::
|
|
|
|
|
|
|
|
|
|
rank\_loss &= max(0, -label * (left - right) + margin)
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
label (Variable): Indicats whether left higher than (right + margin) or not.
|
|
|
|
|
left (Variable): rank score for left.
|
|
|
|
|
right (Variable): rank score for right.
|
|
|
|
|
margin (float): Indicates the margin to be added to right
|
|
|
|
|
label (Variable): Indicates whether the left is ranked higher than the right or not.
|
|
|
|
|
left (Variable): ranking score for left.
|
|
|
|
|
right (Variable): ranking score for right.
|
|
|
|
|
margin (float): Indicates the given margin to be added to right
|
|
|
|
|
name (str|None): A name for this layer (optional). If set None, the layer
|
|
|
|
|
will be named automatically.
|
|
|
|
|
Returns:
|
|
|
|
|
list: The value of rank loss.
|
|
|
|
|
list: The Variable of ranking loss.
|
|
|
|
|
Raises:
|
|
|
|
|
ValueError: Any of label, left, and right is not a variable.
|
|
|
|
|
ValueError: Any of label, left, and right is not a Variable.
|
|
|
|
|
Examples:
|
|
|
|
|
.. code-block:: python
|
|
|
|
|
label = fluid.layers.data(name="label", shape=[4, 1], dtype="float32")
|
|
|
|
@ -5862,8 +5863,8 @@ def margin_rank_loss(label, left, right, margin=0.1, name=None):
|
|
|
|
|
raise ValueError("The left should be a Variable")
|
|
|
|
|
if not (isinstance(right, Variable)):
|
|
|
|
|
raise ValueError("The right should be a Variable")
|
|
|
|
|
out = helper.create_tmp_variable("float32")
|
|
|
|
|
act = helper.create_tmp_variable("float32")
|
|
|
|
|
out = helper.create_tmp_variable(left.dtype)
|
|
|
|
|
act = helper.create_tmp_variable(left.dtype)
|
|
|
|
|
helper.append_op(
|
|
|
|
|
type='margin_rank_loss',
|
|
|
|
|
inputs={"Label": label,
|
|
|
|
|