mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-06-08 15:05:17 +02:00
update model list
This commit is contained in:
parent
1589a04cdb
commit
788e42ea55
1 changed files with 2 additions and 2 deletions
|
|
@ -35,8 +35,8 @@ The current task is about feature engineering. when performing it, please adhere
|
|||
MODEL_TRAIN_PROMPT = """
|
||||
The current task is about training a model, please ensure high performance:
|
||||
- For tabular datasets - you have access to XGBoost, CatBoost, random forest, extremely randomized trees, k-nearest neighbors, linear regression, etc.
|
||||
- For image datasets - you have access to ResNet, VGG, Inception, MobileNet, DenseNet, EfficientNet, etc.
|
||||
- For text datasets - you have access to BERT, GPT-2, RoBERTa, DistilBERT, T5, etc.
|
||||
- For image datasets - you have access to Swin Transformer, ViT, ResNet, EfficientNet, etc.
|
||||
- For text datasets - you have access to Electra, DeBERTa, GPT-2, BERT, etc.
|
||||
- Avoid the use of SVM because of its high training time.
|
||||
- Keep in mind that your user prioritizes results and is highly focused on model performance. So, when needed, feel free to use models of any complexity to improve effectiveness, such as XGBoost, CatBoost, etc.
|
||||
- If non-numeric columns exist, perform label encode together with all steps.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue