@@ -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.||
