项目简介
本项目是基于PyTorch框架开发的依存句法分析树库系统,用于自然语言处理领域的依存句法分析任务。系统定义了双向树LSTM模型,具备解析器、训练、评估和解码等功能。
项目的主要特性和功能
- 双向树LSTM模型:能同时考虑节点的子节点和父节点信息。
- 解析器:可处理依存句法分析任务,包含创建字典、加载和保存模型、训练、评估和解码预测结果等步骤。
- 损失函数:可计算不同类型的损失函数值,用于模型训练。
- 优化器:采用PyTorch的Adam优化器和学习率调度器,实现模型参数优化和学习率动态调整。
- 评估指标:提供评估指标类,用于记录和评估模型性能。
安装使用步骤
- 安装依赖:确保安装Python 3.6.5、PyTorch 0.4.1和Cython 0.28.2。
- 配置:在
exp
目录下,通过config.txt
文件设置数据文件、预训练词嵌入地址、损失类型等配置。 - 运行:运行
run.sh
脚本,按需进行训练或测试。若没有字典,先执行创建字典的命令;有字典和初始化模型后,可进行模型训练。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】