From 9f0427838324f90e4bc9ed62c6eba9e6a2ad4465 Mon Sep 17 00:00:00 2001 From: Yizhou Chi Date: Fri, 13 Sep 2024 11:18:15 +0800 Subject: [PATCH] comment --- expo/data/hf_data.py | 35 +++++++++++++++++++---------------- 1 file changed, 19 insertions(+), 16 deletions(-) diff --git a/expo/data/hf_data.py b/expo/data/hf_data.py index 6f1f55f6d..952ab5c73 100644 --- a/expo/data/hf_data.py +++ b/expo/data/hf_data.py @@ -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", + # }, ]