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

remove dependency

Browse source
This commit is contained in:
Yizhou Chi 2024-10-11 13:12:03 +08:00
parent 9c54113f77
commit ae12d73747
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
expo/data/dataset.py
View file

@ -9,6 +9,7 @@ import yaml
from sklearn.model_selection import train_test_split
from expo.insights.solution_designer import SolutionDesigner
from expo.utils import DATA_CONFIG
BASE_USER_REQUIREMENT = """
This is a {datasetname} dataset. Your goal is to predict the target column `{target_col}`.
@ -361,7 +362,7 @@ async def process_dataset(dataset, solution_designer: SolutionDesigner, save_ana
if __name__ == "__main__":
datasets_dir = "D:/work/automl/datasets"
datasets_dir = DATA_CONFIG["datasets_dir"]
force_update = False
save_analysis_pool = True
datasets_dict = {"datasets": {}}

3
expo/data/hf_data.py
View file

@ -9,6 +9,7 @@ from PIL import Image
from expo.data.dataset import ExpDataset, process_dataset, save_datasets_dict_to_yaml
from expo.insights.solution_designer import SolutionDesigner
from expo.utils import DATA_CONFIG
HFDATSETS = [
{"name": "sms_spam", "dataset_name": "ucirvine/sms_spam", "target_col": "label", "modality": "text"},
@ -114,7 +115,7 @@ class HFExpDataset(ExpDataset):
if __name__ == "__main__":
dataset_dir = "D:/work/automl/datasets"
dataset_dir = DATA_CONFIG["datasets_dir"]
save_analysis_pool = True
force_update = False
datasets_dict = {"datasets": {}}