1.2 KiB
1.2 KiB
模型配置说明
当前使用的模型
项目已配置为使用以下两个硅基流动模型：
1. NER（命名实体识别）
- 模型:
deepseek-ai/DeepSeek-R1-Distill-Qwen-7B - 用途: 从法规文本中提取实体（目标要素、原则要素、空间要素等）
- 特点: DeepSeek R1系列，具有推理能力
2. RE（关系抽取）
- 模型1:
deepseek-ai/DeepSeek-R1-Distill-Qwen-7B - 模型2:
Qwen/Qwen2.5-7B-Instruct - 用途: 从文本中提取实体之间的关系
- 特点: 多模型并行，提高准确性和覆盖率
3. 社区摘要生成
- 模型:
deepseek-ai/DeepSeek-R1-Distill-Qwen-7B - 用途: 为知识图谱的社区生成摘要
配置位置
模型配置在以下文件中：
src/extraction/ner.py: NER模型配置src/extraction/re.py: RE模型配置（多模型）src/kg_builder/indexer.py: 社区摘要生成模型配置
如何修改模型
如果要修改使用的模型，请编辑上述文件中的模型名称字符串。
测试模型可用性
运行以下命令测试模型：
uv run python tests/test_model.py
# 测试指定的两个模型
uv run python tests/test_my_models.py