DISC-FinLLM 开源项目使用教程
1. 项目的目录结构及介绍
DISC-FinLLM 项目的目录结构如下:
DISC-FinLLM/
├── README.md
├── LICENSE
├── requirements.txt
├── data/
│ ├── sample_data.csv
│ └── processed_data.csv
├── models/
│ ├── model_config.json
│ └── model_weights.h5
├── scripts/
│ ├── train.py
│ ├── evaluate.py
│ └── predict.py
├── config/
│ ├── default_config.yaml
│ └── custom_config.yaml
└── docs/
├── user_guide.md
└── developer_guide.md
目录结构介绍
README.md
: 项目介绍文档。LICENSE
: 项目许可证文件。requirements.txt
: 项目依赖包列表。data/
: 存放数据文件的目录。models/
: 存放模型配置和权重文件的目录。scripts/
: 存放训练、评估和预测脚本的目录。config/
: 存放配置文件的目录。docs/
: 存放用户和开发者指南文档的目录。
2. 项目的启动文件介绍
项目的启动文件主要是 scripts/train.py
和 scripts/predict.py
。
scripts/train.py
该文件用于训练模型,主要功能包括:
- 加载数据
- 配置模型参数
- 训练模型
- 保存模型权重
scripts/predict.py
该文件用于模型预测,主要功能包括:
- 加载预训练模型
- 读取输入数据
- 进行预测
- 输出预测结果
3. 项目的配置文件介绍
项目的配置文件主要存放在 config/
目录下,包括 default_config.yaml
和 custom_config.yaml
。
config/default_config.yaml
默认配置文件,包含项目的默认参数设置,如:
- 数据路径
- 模型参数
- 训练参数
config/custom_config.yaml
自定义配置文件,用户可以根据需要修改配置参数,如:
- 自定义数据路径
- 调整模型参数
- 修改训练参数
以上是 DISC-FinLLM 开源项目的基本使用教程,包括项目的目录结构、启动文件和配置文件的介绍。希望对您有所帮助。