openprompt

1
dataset:
2
  name: FewNERD
3
  path: datasets/Typing/FewNERD
4
  label_path_sep: '-' # label path separation token
5

6
plm:
7
  model_name: bert
8
  model_path: bert-base-cased
9
  optimize:
10
    freeze_para: False
11
    lr: 0.00003
12
    weight_decay: 0.01
13
    scheduler:
14
      type: 
15
      num_warmup_steps: 500
16

17
train:
18
  batch_size: 8
19

20
test:
21
  batch_size: 16
22

23
dev:
24
  batch_size: 16
25

26

27

28
template: mixed_template
29
verbalizer: manual_verbalizer
30

31
classification:
32
  metric: ['accuracy', 'loose-micro-f1', 'loose-macro-f1']
33

34
mixed_template:
35
  choice: 0
36
  file_path: scripts/Typing/FewNERD/mixed_template.txt
37

38
manual_verbalizer:
39
  choice: 0
40
  file_path: scripts/Typing/FewNERD/fewnerd_verbalizer.json
41
  
42
environment:
43
  num_gpus: 1
44
  cuda_visible_devices:
45
    - 0
46
  local_rank: 0 
47

48
learning_setting: few_shot
49

50
few_shot:
51
  parent_config: learning_setting
52
  few_shot_sampling: sampling_from_train
53
  
54
sampling_from_train:
55
  parent_config: few_shot_sampling
56
  num_examples_per_label: 8
57
  also_sample_dev: True
58
  num_examples_per_label_dev: 8
59
  seed:
60
    - 123
61

62