BERT-IMDB

它是什么？

BERT（bert-large-cased）在IMDB数据集上进行情感分类训练，数据集包含 IMDB dataset 个样本。

训练设置

该模型使用simpletransformers库，在IMDB情感分类数据集的80%上进行了三个epochs的训练，学习率为1e-5。该库使用学习率调度。

结果

该模型在验证集上达到了90%的分类准确率。

参考资料

完整实验结果可在 tlr repo 中找到。