Initial: integrated 2025 LawGraph (graphrag_pipeline) + 2026 kg_project

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-06-17 10:13:39 +08:00
commit 6c1a69af0d
83 changed files with 14295 additions and 0 deletions
@@ -0,0 +1,43 @@
+# 模型配置说明
+
+## 当前使用的模型
+
+项目已配置为使用以下两个硅基流动模型：
+
+### 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`: 社区摘要生成模型配置
+
+## 如何修改模型
+
+如果要修改使用的模型，请编辑上述文件中的模型名称字符串。
+
+## 测试模型可用性
+
+运行以下命令测试模型：
+```bash
+uv run python tests/test_model.py
+
+# 测试指定的两个模型
+uv run python tests/test_my_models.py
+```
+
+