From 883006a018865595ca04056413faa6a6affcee15 Mon Sep 17 00:00:00 2001
From: dongzhihong <dzhwinter@gmail.com>
Date: Wed, 19 Jul 2017 15:17:47 +0800
Subject: [PATCH] "rename infer network name"

---
 ltr/ranknet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ltr/ranknet.py b/ltr/ranknet.py
index 862d5dea..c14a6907 100644
--- 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)))
 
-- 
GitLab