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

add special instruction for img/text dataset

Browse source
This commit is contained in:
Yizhou Chi 2024-09-24 20:58:41 +08:00
parent 6344046c31
commit 31adaee23f
2 changed files with 14 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
expo/data/dataset.py
View file

@ -20,12 +20,23 @@ USE_AG = """
7. Please use autogluon for model training with presets='medium_quality', time_limit=None, give dev dataset to tuning_data, and use right eval_metric.
"""
TEXT_MODALITY = """
7. You could use models from transformers library for this text dataset.
8. Use gpu if available for faster training.
"""
IMAGE_MODALITY = """
7. You could use models from torchvision library for this image dataset.
8. Use gpu if available for faster training.
"""
STACKING = """
7. To avoid overfitting, train a weighted ensemble model such as StackingClassifier or StackingRegressor.
8. You could do some quick model prototyping to see which models work best and then use them in the ensemble.
"""
SPECIAL_INSTRUCTIONS = {"ag": USE_AG, "stacking": STACKING}
SPECIAL_INSTRUCTIONS = {"ag": USE_AG, "stacking": STACKING, "text": TEXT_MODALITY, "image": IMAGE_MODALITY}
DI_INSTRUCTION = """
## Attention