refactor(graphgen): add parameter re_judge to config

Author: ChenZiHong-Gavin

graphgen/configs/config.yaml.example

@@ -13,3 +13,4 @@ traverse_strategy:
   max_tokens: 256
   loss_strategy: only_edge
 web_search: false
+re_judge: false

graphgen/configs/graphgen_config.yaml

@@ -13,3 +13,4 @@ traverse_strategy:
   max_tokens: 256
   loss_strategy: only_edge
 web_search: false
+re_judge: false

graphgen/generate.py

@@ -90,7 +90,7 @@ def main():
 
     graph_gen.quiz(max_samples=config['quiz_samples'])
 
-    graph_gen.judge(re_judge=False)
+    graph_gen.judge(re_judge=config["re_judge"])
 
     graph_gen.traverse()
 

graphgen/models/strategy/travserse_strategy.py

@@ -5,7 +5,7 @@
 
 @dataclass
 class TraverseStrategy(BaseStrategy):
-    # 生成的QA形式：原子、多跳、开放性
+    # 生成的QA形式：原子、多跳、聚合型
     qa_form: str = "atomic" # "atomic" or "multi_hop" or "aggregated"
     # 最大边数和最大token数方法中选择一个生效
     expand_method: str = "max_tokens" # "max_width" or "max_tokens"
@@ -21,8 +21,6 @@ class TraverseStrategy(BaseStrategy):
     edge_sampling: str = "max_loss" # "max_loss" or "min_loss" or "random"
     # 孤立节点的处理策略
     isolated_node_strategy: str = "add" # "add" or "ignore"
-    # 难度顺序 ["easy", "medium", "hard"], ["hard", "medium", "easy"], ["medium", "medium", "medium"]
-    difficulty_order: list = field(default_factory=lambda: ["medium", "medium", "medium"])
     loss_strategy: str = "only_edge"  # only_edge, both
 
     def to_yaml(self):