Oranblock commited on
Commit
0542c32
·
verified ·
1 Parent(s): 99ac4e0

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "model_name": "bert-base-uncased",
3
- "text_column": "text",
4
- "target_column": "label",
5
  "num_train_epochs": 3,
6
  "per_device_train_batch_size": 8,
7
  "per_device_eval_batch_size": 8,
@@ -10,5 +10,5 @@
10
  "learning_rate": 5e-5,
11
  "fp16": false,
12
  "push_to_hub": true,
13
- "hub_model_id": "your_model_name_on_hub"
14
  }
 
1
  {
2
  "model_name": "bert-base-uncased",
3
+ "text_column": "text", # Update this if needed
4
+ "target_column": "correct_label_column", # Update this with the correct label column name
5
  "num_train_epochs": 3,
6
  "per_device_train_batch_size": 8,
7
  "per_device_eval_batch_size": 8,
 
10
  "learning_rate": 5e-5,
11
  "fp16": false,
12
  "push_to_hub": true,
13
+ "hub_model_id": "Oranblock/marblex"
14
  }