From 1589a04cdbf14ef14549bd09141483191b85929b Mon Sep 17 00:00:00 2001 From: Yizhou Chi Date: Sat, 28 Sep 2024 18:27:23 +0800 Subject: [PATCH] clarify prediction saving prompt --- expo/data/dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/expo/data/dataset.py b/expo/data/dataset.py index f2f01f71b..dd4cb4543 100644 --- a/expo/data/dataset.py +++ b/expo/data/dataset.py @@ -26,7 +26,7 @@ TEXT_MODALITY = """ """ IMAGE_MODALITY = """ -7. You could use models from transformers library for this image dataset. +7. You could use models from transformers/torchvision library for this image dataset. 8. Use gpu if available for faster training. """ @@ -50,8 +50,8 @@ DI_INSTRUCTION = """ ## Saving Dev and Test Predictions 1. Save the prediction results of BOTH the dev set and test set in `dev_predictions.csv` and `test_predictions.csv` respectively in the output directory. - Both files should contain a single column named `target` with the predicted values. -2. Make sure the prediction results are in the same format as the target column in the training set. -- For instance, if the target column is categorical, the prediction results should be categorical as well. +2. Make sure the prediction results are in the same format as the target column in the original training set. +- For instance, if the original target column is a list of string, the prediction results should also be strings. ## Output Performance Print the train and dev set performance in the last step.