Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OPTHREE
Paddle
提交
61d98f27
P
Paddle
项目概览
OPTHREE
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
61d98f27
编写于
11月 30, 2017
作者:
Y
Yibing Liu
提交者:
GitHub
11月 30, 2017
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #6058 from kuke/refine_rank_loss_op
Revise comments in rank_loss_op
上级
e1b8c27a
9907350d
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
25 addition
and
16 deletion
+25
-16
paddle/operators/rank_loss_op.cc
paddle/operators/rank_loss_op.cc
+23
-14
paddle/operators/rank_loss_op.cu
paddle/operators/rank_loss_op.cu
+1
-1
paddle/operators/rank_loss_op.h
paddle/operators/rank_loss_op.h
+1
-1
未找到文件。
paddle/operators/rank_loss_op.cc
浏览文件 @
61d98f27
...
...
@@ -4,7 +4,7 @@
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
...
...
@@ -35,9 +35,10 @@ class RankLossOp : public framework::OperatorWithKernel {
auto
right_dims
=
ctx
->
GetInputDim
(
"Right"
);
PADDLE_ENFORCE
((
label_dims
==
left_dims
)
&&
(
left_dims
==
right_dims
),
"All inputs must have the same size"
);
PADDLE_ENFORCE
((
label_dims
.
size
()
==
2
)
&&
(
label_dims
[
1
]
==
1
),
"All inputs must be row vector with size batch_size x 1."
);
"All inputs must have the same size."
);
PADDLE_ENFORCE
(
(
label_dims
.
size
()
==
2
)
&&
(
label_dims
[
1
]
==
1
),
"All inputs must be 2-D tensors with shape [batch_size x 1]."
);
ctx
->
SetOutputDim
(
"Out"
,
label_dims
);
}
};
...
...
@@ -48,10 +49,17 @@ class RankLossOpMaker : public framework::OpProtoAndCheckerMaker {
framework
::
OpAttrChecker
*
op_checker
)
:
OpProtoAndCheckerMaker
(
proto
,
op_checker
)
{
AddInput
(
"Label"
,
"The label indicating A ranked higher than B or not, row vector."
);
AddInput
(
"Left"
,
"The output of RankNet for doc A, vector."
);
AddInput
(
"Right"
,
"The output of RankNet for doc B, vetor."
);
AddOutput
(
"Out"
,
"The output loss of RankLoss operator, vector."
);
"(2-D Tensor with shape [batch_size x 1]) "
"The label indicating A ranked higher than B or not."
);
AddInput
(
"Left"
,
"(2-D Tensor with shape [batch_size x 1]) "
"The output of RankNet for doc A."
);
AddInput
(
"Right"
,
"(2-D Tensor with shape [batch_size x 1]) "
"The output of RankNet for doc B."
);
AddOutput
(
"Out"
,
"(2-D Tensor with shape [batch_size x 1]) "
"The output loss of RankLoss operator."
);
AddComment
(
R"DOC(
RankLoss Operator.
...
...
@@ -65,16 +73,17 @@ P = {0, 1} or {0, 0.5, 1}, where 0.5 means no information about the rank of
the input pair.
The RankLoss operator takes three inputs: Left (o_i), Right (o_j) and Label
(P_{i,j}), which represent the output of RankNet for the two docs and the label,
respectively, and yields the rank loss C_{i,j} using the following equation:
(P_{i,j}), which represent the output score of RankNet for the two docs and
the label respectively, and yields the rank loss C_{i,j} using the following
equation:
\f
$$
C_{i,j} = -\tilde{P_{ij}} * o_{i,j} + log(1 + e^{o_{i,j}}) \\
$$
C_{i,j} = -\tilde{P_{ij}} * o_{i,j} +
\
log(1 + e^{o_{i,j}}) \\
o_{i,j} = o_i - o_j \\
\tilde{P_{i,j}} = \left \{0, 0.5, 1 \right \} \ or \ \left \{0, 1 \right \}
\f
$$
$$
The operator can take
inputs of one sample or in batch
.
The operator can take
batch inputs with size batch_size (batch_size >= 1)
.
)DOC"
);
}
...
...
paddle/operators/rank_loss_op.cu
浏览文件 @
61d98f27
...
...
@@ -4,7 +4,7 @@
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
...
...
paddle/operators/rank_loss_op.h
浏览文件 @
61d98f27
...
...
@@ -4,7 +4,7 @@
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录