OptimizersForTextClassfication/README.md

588 B
Raw Blame History

不同优化算法对BERT情感分类的影响

来源北京理工大学CSTAIG研究组人工智能练习

使用方法

主要依赖库:

  • Pytorch (>=1.11.0)
  • transformers
  • tqdm
  • csv

  1. 下载数据集: https://pan.baidu.com/s/19tjsb6R5Czs4UDmm_xWQOg 提取码: 33cw 存放目录与Model.py等文件相同即可也可不同但需要对源代码略作修改
  2. 使用IDE或命令行运行Dataset.py,处理数据集
  3. 使用IDE或命令行运行Model.py,进行测试
  4. 如需执行更多个性化操作,可以查看源代码