Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OPTHREE
Paddle
提交
6fed6f20
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,发现更多精彩内容 >>
提交
6fed6f20
编写于
11月 20, 2017
作者:
W
wangmeng28
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add support of sparse_binary_vector as input for fm layer
上级
5392a503
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
34 addition
and
13 deletion
+34
-13
paddle/gserver/layers/FactorizationMachineLayer.cpp
paddle/gserver/layers/FactorizationMachineLayer.cpp
+14
-6
paddle/gserver/layers/FactorizationMachineLayer.h
paddle/gserver/layers/FactorizationMachineLayer.h
+1
-0
paddle/math/CpuSparseMatrix.cpp
paddle/math/CpuSparseMatrix.cpp
+19
-7
未找到文件。
paddle/gserver/layers/FactorizationMachineLayer.cpp
浏览文件 @
6fed6f20
...
...
@@ -96,15 +96,20 @@ void FactorizationMachineLayer::backward(const UpdateCallback& callback) {
/* Calculate the gradients of the latentVectors_ matrix */
if
(
latentVectors_
->
getWGrad
())
{
MatrixPtr
tmpInput
=
inputV
->
clone
(
0
,
0
,
useGpu_
);
if
(
dynamic_cast
<
CpuSparseMatrix
*>
(
inputV
.
get
()))
{
Matrix
::
resizeOrCreateSparseMatrix
(
tmpInput_
,
inputV
->
getHeight
(),
inputV
->
getWidth
(),
inputV
->
getElementCnt
());
CpuSparseMatrix
*
sparseInputV
=
dynamic_cast
<
CpuSparseMatrix
*>
(
inputV
.
get
());
CpuSparseMatrix
*
sparseInputSquare
=
dynamic_cast
<
CpuSparseMatrix
*>
(
inputSquare_
.
get
());
CpuSparseMatrix
*
sparseTmpInput
=
dynamic_cast
<
CpuSparseMatrix
*>
(
tmpInput
.
get
());
dynamic_cast
<
CpuSparseMatrix
*>
(
tmpInput
_
.
get
());
sparseTmpInput
->
copyFrom
(
*
sparseInputV
);
sparseTmpInput
->
rowScale
(
0
,
*
sparseInputV
,
*
oGrad
);
latentVectors_
->
getWGrad
()
->
mul
(
*
sparseTmpInput
->
getTranspose
(),
*
inputMulFactor_
,
1
,
1
);
...
...
@@ -115,12 +120,15 @@ void FactorizationMachineLayer::backward(const UpdateCallback& callback) {
negOnes_
->
add
(
-
1
);
tmpSum_
->
mul
(
*
negOnes_
,
*
sparseTmpInput
,
1
,
0
);
}
else
{
tmpInput
->
rowScale
(
0
,
*
inputV
,
*
oGrad
);
Matrix
::
resizeOrCreate
(
tmpInput_
,
inputV
->
getHeight
(),
inputV
->
getWidth
(),
false
,
useGpu_
);
tmpInput_
->
rowScale
(
0
,
*
inputV
,
*
oGrad
);
latentVectors_
->
getWGrad
()
->
mul
(
*
tmpInput
->
getTranspose
(),
*
inputMulFactor_
,
1
,
1
);
tmpInput
->
rowScale
(
0
,
*
inputSquare_
,
*
oGrad
);
*
tmpInput
_
->
getTranspose
(),
*
inputMulFactor_
,
1
,
1
);
tmpInput
_
->
rowScale
(
0
,
*
inputSquare_
,
*
oGrad
);
tmpSum_
->
sumCols
(
*
tmpInput
,
-
1
,
0
);
tmpSum_
->
sumCols
(
*
tmpInput
_
,
-
1
,
0
);
}
latentVectors_
->
getWGrad
()
->
addRowScale
(
...
...
paddle/gserver/layers/FactorizationMachineLayer.h
浏览文件 @
6fed6f20
...
...
@@ -61,6 +61,7 @@ private:
// Store temporary calculation result
MatrixPtr
tmpOut_
;
MatrixPtr
tmpSum_
;
MatrixPtr
tmpInput_
;
// Negative identity matrix
MatrixPtr
negOnes_
;
...
...
paddle/math/CpuSparseMatrix.cpp
浏览文件 @
6fed6f20
...
...
@@ -266,6 +266,7 @@ void CpuSparseMatrix::rowScale(size_t cCol, CpuSparseMatrix& b, Matrix& c) {
CHECK_EQ
(
width_
,
b
.
getWidth
());
real
*
A
=
getValue
();
real
*
B
=
b
.
getValue
();
if
(
b
.
getValueType
()
==
FLOAT_VALUE
)
{
for
(
size_t
i
=
0
;
i
<
height_
;
i
++
)
{
size_t
start
=
getRowStartIdx
(
i
);
size_t
end
=
getRowStartIdx
(
i
+
1
);
...
...
@@ -275,6 +276,17 @@ void CpuSparseMatrix::rowScale(size_t cCol, CpuSparseMatrix& b, Matrix& c) {
A
[
j
]
=
B
[
j
]
*
c
.
getElement
(
i
,
cCol
);
}
}
}
else
if
(
b
.
getValueType
()
==
NO_VALUE
)
{
for
(
size_t
i
=
0
;
i
<
height_
;
i
++
)
{
size_t
start
=
getRowStartIdx
(
i
);
size_t
end
=
getRowStartIdx
(
i
+
1
);
CHECK_EQ
(
start
,
b
.
getRowStartIdx
(
i
));
CHECK_EQ
(
end
,
b
.
getRowStartIdx
(
i
+
1
));
for
(
size_t
j
=
start
;
j
<
end
;
j
++
)
{
A
[
j
]
=
c
.
getElement
(
i
,
cCol
);
}
}
}
}
void
CpuSparseMatrix
::
randomizeUniform
()
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录