diff --git a/README.md b/README.md index cb95f7dd23b90e5ad8aada8a32377b11087da7a0..540670bf063206e418e6811c461cbd41583213d4 100644 --- a/README.md +++ b/README.md @@ -184,7 +184,7 @@ For `<COMMAND>` you must enter one of the commands you find in the list below, w | Command | Functionality | Arguments | | ------- | ------------- |-----------| -|**General**| +| <center> **General** </center>| |**`--architecture`** | Defines which model is used. | Choose `bert-base-uncased` or `roberta` | |**`--model_type`** | How to initialize the Classification Model | Choose `separate` or `one` | |**`--mixlayer`**| Specify in which `layer` the interpolation takes place. Only select one layer at a time. | Choose from ${0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11}$ | @@ -199,12 +199,12 @@ For `<COMMAND>` you must enter one of the commands you find in the list below, w |**`-tc`**/**`--tcontext`**|Whether or not to preprocess the training set with context.|| |**`--masking`**|Whether or not to mask the target word.|| |**`-lambda`**/**`--lambda_value`**|Speficies the lambda value for interpolation of *MixUp* and *TMix*|Default is $0.4$, `type=float`| -|**MixUp** specific| +| <center>| **MixUp** specific </center>|| |**`-mixup`**/**`--mix_up`**| Whether or not to use *MixUp*. If yes, please specify `lambda` and `-mixepoch`| | |**`-mixepoch`**/**`--mixepoch`**|Specifies the epoch(s) in which to apply *MixUp*.|Default is `None`| -|**TMix** specific| -|**`--tmix`**| Whether or not to use *TMix*. If yes, please specify `layer` and `-lambda`| | -|**Datasets** specific| +| <center>| **TMix** specific </center>|| +|**`--tmix`**| Whether or not to use *TMix*. If yes, please specify `-mixlayer` and `-lambda`| | +| <center>| **Datasets** specific </center>|| |**`-t`**/**`"--train_dataset`**|Defines which dataset is chosen for training.|Choose any of the datasets from [original_datasets](data/original_datasets), [fused_datasets](data/fused_datasets) or [paraphrases](data/paraphrases)| |**`-v`**/**`--test_dataset`**|Defines which dataset is chosen for testing.|Choose from ["semeval_test.txt"](data/original_datasets/semeval_test.txt), ["companies_test.txt"](data/original_datasets/companies_test.txt) or ["relocar_test.txt"](data/original_datasets/relocar_test.txt)| |**`--imdb`**| Whether or not to use the [IMDB](https://huggingface.co/datasets/imdb) dataset. Note that this is only relevant for validating our *TMix* implementation.||