| python run_classification.py \ | |
| --model_name_or_path bert-base-uncased \ | |
| --train_file data/train.csv \ | |
| --validation_file data/val.csv \ | |
| --test_file data/test.csv \ | |
| --shuffle_train_dataset \ | |
| --metric_name evaluate/metrics/accuracy/accuracy.py \ | |
| --text_column_name "text" \ | |
| --text_column_delimiter "\n" \ | |
| --label_column_name "target" \ | |
| --do_train \ | |
| --do_eval \ | |
| --do_predict \ | |
| --max_seq_length 512 \ | |
| --per_device_train_batch_size 32 \ | |
| --learning_rate 2e-5 \ | |
| --num_train_epochs 1 \ | |
| --output_dir experiments/ |