Update README.md
Browse files
README.md
CHANGED
@@ -12,12 +12,21 @@ tags: []
|
|
12 |
![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/66790f78c411cf3d84e7ceb0/owIYJZH3FK9VgUfloZ2Rc.jpeg)
|
13 |
## ---------
|
14 |
checkpoint = "mental/mental-bert-base-uncased"
|
|
|
15 |
learning_rate = 2e-5
|
|
|
16 |
batch_size = 16
|
|
|
17 |
num_train_epochs = 5
|
|
|
18 |
weight_decay = 0.01
|
|
|
19 |
metric_name = "accuracy"
|
|
|
20 |
optimizer = AdamW
|
|
|
21 |
test_size = 0.2
|
|
|
22 |
logging_steps = 25
|
|
|
23 |
column_item = "labels_array"
|
|
|
12 |
![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/66790f78c411cf3d84e7ceb0/owIYJZH3FK9VgUfloZ2Rc.jpeg)
|
13 |
## ---------
|
14 |
checkpoint = "mental/mental-bert-base-uncased"
|
15 |
+
|
16 |
learning_rate = 2e-5
|
17 |
+
|
18 |
batch_size = 16
|
19 |
+
|
20 |
num_train_epochs = 5
|
21 |
+
|
22 |
weight_decay = 0.01
|
23 |
+
|
24 |
metric_name = "accuracy"
|
25 |
+
|
26 |
optimizer = AdamW
|
27 |
+
|
28 |
test_size = 0.2
|
29 |
+
|
30 |
logging_steps = 25
|
31 |
+
|
32 |
column_item = "labels_array"
|