参数名称 | 参数值 |
epochs | 100 |
batch_size | 64 |
学习率 | 1e−5 |
dropout | 0.9 |
GRU隐藏层维度 | 768 |
损失函数 | Cross-Entropy |
优化器 | BertAdam |
LDA最大主题数 | 20 |
LDA最大迭代次数 | 100 |
Learning_method | batch |