Merge pull request #171 from dzhwinter/fix_ltr

"rename infer network name"

Merge pull request #171 from dzhwinter/fix_ltr
"rename infer network name"
3a784aa0 · dzhwinter · GitHub · 63fc48b0 · 883006a0 · 3a784aa0
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ltr/ranknet.py ltr/ranknet.py +1 -1

未找到文件。
--- a/ltr/ranknet.py
+++ b/ltr/ranknet.py
@@ -102,7 +102,7 @@ def ranknet_infer(pass_id):
    feature_dim = 46
    # 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(
        gzip.open("ranknet_params_%d.tar.gz" % (pass_id - 1)))