eperezs commited on
Commit
4c649b3
1 Parent(s): 9695c8f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -0
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"