# pytorch-NLP **Repository Path**: CIPFZ/pytorch-nlp ## Basic Information - **Project Name**: pytorch-NLP - **Description**: 基于pytorch的自然语言相关代码 - **Primary Language**: Python - **License**: Not specified - **Default Branch**: master - **Homepage**: None - **GVP Project**: No ## Statistics - **Stars**: 0 - **Forks**: 0 - **Created**: 2022-11-29 - **Last Updated**: 2023-09-30 ## Categories & Tags **Categories**: Uncategorized **Tags**: None ## README # pytorch-NLP #### 介绍 1、Bert+TextCNN 实现新闻文本单分类 #### 项目路径 1、bert_TextCNN 新闻单分类 #### 安装教程 1. 将代码全部下载 2. 安装相应的库(在requirments.txt文件目录下) ``` pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple ``` #### 使用说明 2. **machine-learn / bert_TextCNN 新闻单分类 使用说明:** #### 文件介绍: 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 为配置文件 根据需要进行修改 ``` #### 注意 1. 本代码只是用于交流学习,不支持商业交易活动。如有疑问和指教请给我留言,我会尽快回复。 2. 转载请附上我的链接:https://gitee.com/CIPFZ/machine-learn/tree/master/CNN/%E7%8C%AB%E7%8B%97%E5%A4%A7%E6%88%98 3. 谢谢配合!