gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-06 19:47:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: enable to inject diversity cross async multi-trace (#1173)

Browse Source 
* init multi_trace_async_ diversity inject

* lint

* add task in context for diversity injection, add abstract class for diversity injection

* lint

* feat: update diversity injection strategy and enhance sibling context handling in experiment generation

* add always inject

---------

Co-authored-by: Xu Yang <peteryang@vip.qq.com>
This commit is contained in:
xuangu-fang
2025-08-18 12:43:21 +08:00
committed by GitHub
parent 0885d4d218
commit ceabfe9686
6 changed files with 218 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/app/data_science/conf.py
+10
View File
@@ -111,6 +111,16 @@ class DataScienceBasePropSetting(KaggleBasePropSetting):
# inject diverse when start a new sub-trace
enable_inject_diverse: bool = False
# inject diverse from other traces when start a new sub-trace
enable_cross_trace_diversity: bool = True
"""Enable cross-trace diversity injection when starting a new sub-trace.
This is different from `enable_inject_diverse` which is for non-parallel cases."""
diversity_injection_strategy: str = (
"rdagent.scenarios.data_science.proposal.exp_gen.diversity_strategy.InjectUntilSOTAGainedStrategy"
)
"""The strategy to use for injecting diversity context."""
# enable different version of DSExpGen for multi-trace
enable_multi_version_exp_gen: bool = False
exp_gen_version_list: str = "v3,v2"