Update config.json
Browse files- config.json +1 -1
config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"model_name": "bert-base-uncased",
|
| 3 |
-
"
|
| 4 |
"target_column": "command",
|
| 5 |
"num_train_epochs": 3,
|
| 6 |
"per_device_train_batch_size": 8,
|
|
|
|
| 1 |
{
|
| 2 |
"model_name": "bert-base-uncased",
|
| 3 |
+
"text_columns": ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11"], # List all input columns
|
| 4 |
"target_column": "command",
|
| 5 |
"num_train_epochs": 3,
|
| 6 |
"per_device_train_batch_size": 8,
|