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

comment

Browse source
This commit is contained in:
Yizhou Chi 2024-09-13 11:18:15 +08:00
parent 562af8c4e4
commit 9f04278383
1 changed files with 19 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

35
expo/data/hf_data.py
View file

@ -9,22 +9,25 @@ from expo.data.dataset import ExpDataset, process_dataset, save_datasets_dict_to
from expo.insights.solution_designer import SolutionDesigner
HFDATSETS = [
# {"name": "sms_spam", "dataset_name": "ucirvine/sms_spam", "target_col": "label", "modality": "text"},
# {"name": "banking77", "dataset_name": "PolyAI/banking77", "target_col": "label", "modality": "text"},
# {"name": "gnad10", "dataset_name": "community-datasets/gnad10", "target_col": "label", "modality": "text"},
{
"name": "oxford-iiit-pet",
"dataset_name": "timm/oxford-iiit-pet",
"target_col": "label_cat_dog",
"modality": "image",
},
{"name": "stanford_cars", "dataset_name": "tanganke/stanford_cars", "target_col": "label", "modality": "image"},
{
"name": "fashion_mnist",
"dataset_name": "zalando-datasets/fashion_mnist",
"target_col": "label",
"modality": "image",
},
{"name": "sms_spam", "dataset_name": "ucirvine/sms_spam", "target_col": "label", "modality": "text"},
{"name": "banking77", "dataset_name": "PolyAI/banking77", "target_col": "label", "modality": "text"},
{"name": "gnad10", "dataset_name": "community-datasets/gnad10", "target_col": "label", "modality": "text"},
# {
# "name": "oxford-iiit-pet",
# "dataset_name": "timm/oxford-iiit-pet",
# "target_col": "label_cat_dog",
# "modality": "image",
# },
# { "name": "stanford_cars",
# "dataset_name": "tanganke/stanford_cars",
# "target_col": "label",
# "modality": "image"},
# {
# "name": "fashion_mnist",
# "dataset_name": "zalando-datasets/fashion_mnist",
# "target_col": "label",
# "modality": "image",
# },
]