pytorch-NLP

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

    1. config.py - 项目文件相关的配置
    2. main.py - 运行主文件
    3. test.py - 模型验证代码并模型计算准确率
    4. train.py - 模型训练函数
    5. pridect.py - 新闻文本分类预测函数
    6. DataLoad文件夹 - 根据需要加载训练、验证、测试数据
    7. bert文件夹 - bert模型文件（由于模型太大，需要自行下载： https://huggingface.co/hfl/chinese-bert-wwm-ext/tree/main）下载文件 pytorch_moudel.bin文件即可
    8. data文件夹 - 存放训练、验证等数据
    9. modle文件夹 - TextCNN 模型文件及模型配置文件
    10. output文件夹 - 训练后的模型存放路径
    11. utils 文件夹 - 工具文件夹
            

#### 使用时需要改动的内容：
  每个文件都有相应的注释，根据自己的需要进行更改

        注：main.py 为运行总文件夹， 根据需要运行相关代码
        需要下载Bert模型到bert文件夹下
        config.py 为配置文件 根据需要进行修改