提交 c43d3d26 编写于 作者: Z zhangwenhui03

add ssr & gru4rec bpr

上级 ff0bd6f5
...@@ -30,7 +30,8 @@ def parse_args(): ...@@ -30,7 +30,8 @@ def parse_args():
'--batch_size', type=int, default='50', help='batch_size') '--batch_size', type=int, default='50', help='batch_size')
parser.add_argument( parser.add_argument(
'--hid_size', type=int, default='128', help='hidden size') '--hid_size', type=int, default='128', help='hidden size')
parser.add_argument('--emb_size', type=int, default='128', help='emb size') parser.add_argument(
'--emb_size', type=int, default='128', help='embedding size')
args = parser.parse_args() args = parser.parse_args()
return args return args
......
...@@ -34,10 +34,7 @@ def parse_args(): ...@@ -34,10 +34,7 @@ def parse_args():
parser.add_argument( parser.add_argument(
"--base_lr", type=float, default=0.01, help="learning rate") "--base_lr", type=float, default=0.01, help="learning rate")
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')
parser.add_argument( parser.add_argument(
"--epochs", type=int, default=10, help="Number of epochs") "--epochs", type=int, default=10, help="Number of epochs")
parser.add_argument( parser.add_argument(
...@@ -50,7 +47,8 @@ def parse_args(): ...@@ -50,7 +47,8 @@ def parse_args():
'--model_dir', type=str, default='model_output', help='model dir') '--model_dir', type=str, default='model_output', help='model dir')
parser.add_argument( parser.add_argument(
"--hidden_size", type=int, default=128, help="hidden size") "--hidden_size", type=int, default=128, help="hidden size")
parser.add_argument("--batch_size", type=int, default=50, help="batch size") parser.add_argument(
"--batch_size", type=int, default=50, help="number of batch")
parser.add_argument( parser.add_argument(
"--embedding_dim", type=int, default=128, help="embedding dim") "--embedding_dim", type=int, default=128, help="embedding dim")
parser.add_argument( parser.add_argument(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册