提交 1337e3fb 编写于 作者: C Cao Ying 提交者: GitHub

Merge pull request #222 from lcy-seso/fix_mse_cost

rename mse into square_error after dev branch update.
......@@ -100,6 +100,6 @@ class CTRmodel(object):
self.output = layer.fc(
input=merge_layer, size=1, act=paddle.activation.Sigmoid())
if not self.is_infer:
self.train_cost = paddle.layer.mse_cost(
self.train_cost = paddle.layer.square_error_cost(
input=self.output, label=self.click)
return self.output
......@@ -270,7 +270,7 @@ class DSSM(object):
input=prediction, label=label)
else:
prediction = paddle.layer.cos_sim(*semantics)
cost = paddle.layer.mse_cost(prediction, label)
cost = paddle.layer.square_error_cost(prediction, label)
if not self.is_infer:
return cost, prediction, label
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册