提交 3a784aa0 编写于 作者: D dzhwinter 提交者: GitHub

Merge pull request #171 from dzhwinter/fix_ltr

"rename infer network name"
...@@ -102,7 +102,7 @@ def ranknet_infer(pass_id): ...@@ -102,7 +102,7 @@ def ranknet_infer(pass_id):
feature_dim = 46 feature_dim = 46
# we just need half_ranknet to predict a rank score, which can be used in sort documents # we just need half_ranknet to predict a rank score, which can be used in sort documents
output = half_ranknet("left", feature_dim) output = half_ranknet("infer", feature_dim)
parameters = paddle.parameters.Parameters.from_tar( parameters = paddle.parameters.Parameters.from_tar(
gzip.open("ranknet_params_%d.tar.gz" % (pass_id - 1))) gzip.open("ranknet_params_%d.tar.gz" % (pass_id - 1)))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册