参数名

含义

参数取值

batch_size

每批训练大小

64

num_epochs

最大训练轮数

20

learning_rate

学习率

1e-3

embedding_dim

词向量维度

128

vocab_size

词汇表大小

5000