gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-28 16:07:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: refine merge (#842)

Browse Source 
* feat: add search_type param and customize success trial desc

* docs: simplify trial descriptions in share.yaml

* lint
This commit is contained in:
you-n-g
2025-05-04 23:17:58 +08:00
committed by GitHub
parent 93303873d0
commit 6b7988fba1
2 changed files with 9 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/scenarios/data_science/proposal/exp_gen/merge.py
+7 -2
View File
@@ -37,10 +37,15 @@ class MergeExpGen(ExpGen):
heading="A solution that to be merged into previous best solution",
)
success_fb_list = trace.experiment_and_feedback_list_after_init(return_type="sota", selection=(leaves[1],))
success_fb_list = trace.experiment_and_feedback_list_after_init(
return_type="sota", search_type="ancestors", selection=(leaves[1],)
)
if len(success_fb_list) > 0:
exp_to_merge_fb_desc = T("scenarios.data_science.share:describe.trace").r(
exp_and_feedback_list=success_fb_list, type="success", heading="Successful iterations:"
exp_and_feedback_list=success_fb_list,
type="success",
heading="Successful iterations:",
success_trial_desc="These trials are the steps or changes that led to the success of the solution to be merged",
)
else:
exp_to_merge_fb_desc = T("scenarios.data_science.share:describe.feedback").r(
rdagent/scenarios/data_science/share.yaml
+2 -10
View File
@@ -31,7 +31,6 @@ describe: # some template to describe some object
feedback: |-
{% if exp_and_feedback and exp_and_feedback|length > 1 %}
## {{heading | default('Previous trial and feedback')}}
Before current trial, a previous recent trial is listed below.
{% if exp_and_feedback[0].hypothesis %}
the experiment is designed based on hypothesis: {{ exp_and_feedback[0].hypothesis }}
{% endif %}
@@ -51,16 +50,9 @@ describe: # some template to describe some object
## {{ heading | default('Trace of all trials') }}
{% endif %}
Before current trial, several
Before current trial, several {% if type == "success" %}successful{% elif type == "failure" %}failed{% else %}successful or failed{% endif %} trials are listed below.
{% if type == "success" %}
successful
{% elif type == "failure" %}
failed
{% else %}
successful or failed
{% endif %} trials are listed below.
{% if type == "success" %}
The current SOTA method is the combination of the best solutions of these trials.
{{success_trial_desc | default('The current SOTA method is the combination of the best solutions of these trials.')}}
{% endif %}
The trace order is from the earliest to the latest. Please focus more on the later trials.