apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-15 11:02:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix jinyu's review and add reproduce's code in experiment folder

Browse source
This commit is contained in:
didi 2024-10-24 18:52:37 +08:00
parent d2f90dbda0
commit 39ae68225b
11 changed files with 399 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/ext/aflow/data/download_data.py
View file

@ -55,7 +55,11 @@ datasets_to_download: Dict[str, Dict[str, str]] = {
"filename": "aflow_data.tar.gz",
"extract_path": "metagpt/ext/aflow/data",
},
"results": {"url": "", "filename": "result.tar.gz", "extract_path": "metagpt/ext/aflow/data/results"}, # 请填入正确的URL
"results": {
"url": "",
"filename": "result.tar.gz",
"extract_path": "metagpt/ext/aflow/data/results"
},
"initial_rounds": {
"url": "https://drive.google.com/uc?export=download&id=1UBoW4WBWjX2gs4I_jq3ALdXeLdwDJMdP",
"filename": "initial_rounds.tar.gz",

3
metagpt/ext/aflow/scripts/optimizer_utils/data_utils.py
View file

@ -52,7 +52,8 @@ class DataUtils:
scores = [item["score"] * 100 for item in sorted_items]
probabilities = self._compute_probabilities(scores)
logger.info("\nMixed probability distribution: ", probabilities)
logger.info(f"\nMixed probability distribution: {probabilities}")
logger.info(f"\nSorted rounds: {sorted_items}")
selected_index = np.random.choice(len(sorted_items), p=probabilities)
logger.info(f"\nSelected index: {selected_index}, Selected item: {sorted_items[selected_index]}")