提交 57e6a667 编写于 作者: Z zhangwenhui03

fix style2

上级 c34bd511
......@@ -26,7 +26,8 @@ def parse_args():
'--hid_size', type=int, default=100, help='hidden-dim size')
parser.add_argument(
'--neg_size', type=int, default=10, help='neg item size')
parser.add_argument('--loss', type=str, default="bpr", help='loss fuction')
parser.add_argument(
'--loss', type=str, default="bpr", help='loss: bpr/cross_entropy')
parser.add_argument(
'--model_dir', type=str, default='model_bpr_recall20', help='model dir')
parser.add_argument(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册