pytorch_bert_elmo_example icon indicating copy to clipboard operation
pytorch_bert_elmo_example copied to clipboard

Published 20 hours ago • verified publisher icon ShomyLiu

关于最终训练结果的展示

Open 14H034160212 opened this issue 4 years ago • 1 comments

您好， 感谢您提供这么好的学习资料！我用了ELMO+TextCNN的组合跑了IMDB数据集，得到了下面的结果，想问一下，最后的准确率的计算是应该只算最后一个epoch的accuracy还是把所有epoch的accuracy取平均？另外这个accuracy应该是测试集上的accuracy是吗？因为运行完模型以后没有保存训练的模型，如果能把训练的模型保存下来就更好了。 微信截图_20200701012808

14H034160212 avatar Jun 30 '20 13:06 14H034160212

你好，输出的acc都是test dataset上，可见： https://github.com/ShomyLiu/pytorch_elmo_example/blob/master/main.py#L65 每个epoch都会输出当前模型下在test 数据集上的acc结果。 后续会完善save/load模型

ShomyLiu avatar Jul 01 '20 01:07 ShomyLiu