ABSA-PyTorch 使用 PyTorch 实现的基于方面的情感分析开源项目

我要开发同款
匿名用户2021年10月17日
57阅读
开发技术Python
所属分类人工智能、机器学习/深度学习
授权协议MIT

作品详情

ABSA-PyTorch是一个基于方面的情感分析,使用PyTorch实现。

要求pytorch>=0.4.0numpy>=1.13.3sklearnpython3.6/3.7transformers要安装要求内容,请运行pipinstall-rrequirements.txt.

对于非基于BERT的模型, 需要 GloVe 预训练的词向量,更多细节请参考 data_utils.py。用法训练

pythontrain.py--model_namebert_spc--datasetrestaurant所有实现的模型都列在模型目录中。有关更多训练参数,请参阅 train.py。有关k-fold交叉验证支持,请参阅train_k_fold_cross_val.py。推理

对于非基于BERT的模型和基于BERT的模型,请参阅 infer_example.py。提示

对于非基于BERT的模型,训练过程不是很稳定。基于BERT的模型对小数据集上的超参数(尤其是学习率)更敏感,请参阅此问题。对特定任务进行微调是释放BERT真正威力所必需的。框架

对于灵活的训练/推理和方面术语提取,请尝试 PyABSA,它包含此存储库中的所有模型。

 

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论