Skip to content

M
models

PaddlePaddle / models
大约 2 年 前同步成功

通知 232
Star 6828
Fork 2962
M
models
提交 eb838cae 编写于 作者: Z zhangwenhui03
浏览文件
操作

fix style4

上级 f89d9893
显示空白变更内容
内联 并排
Showing with 3 addition and 5 deletion
fluid/PaddleRec/gru4rec/infer_bpr.py
浏览文件 @ eb838cae
...@@ -19,7 +19,8 @@ def parse_args(): ...@@ -19,7 +19,8 @@ def parse_args():
'--test_dir', type=str, default='test_data', help='test file address') '--test_dir', type=str, default='test_data', help='test file address')
parser.add_argument( parser.add_argument(
'--start_index', type=int, default='1', help='start index') '--start_index', type=int, default='1', help='start index')
parser.add_argument('--last_index', type=int, default='3', help='end index') parser.add_argument(
'--last_index', type=int, default='3', help='last index')
parser.add_argument( parser.add_argument(
'--model_dir', type=str, default='model_bpr_recall20', help='model dir') '--model_dir', type=str, default='model_bpr_recall20', help='model dir')
parser.add_argument( parser.add_argument(
...@@ -29,10 +30,7 @@ def parse_args(): ...@@ -29,10 +30,7 @@ def parse_args():
parser.add_argument( parser.add_argument(
'--hid_size', type=int, default='100', help='batch_size') '--hid_size', type=int, default='100', help='batch_size')
parser.add_argument( parser.add_argument(
'--vocab_path', '--vocab_path', type=str, default='vocab.txt', help='vocab file')
type=str,
default='vocab.txt',
help='vocab file address')
args = parser.parse_args() args = parser.parse_args()
return args return args
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或