27 lines
416 B
INI
27 lines
416 B
INI
|
[DEFAULT]
|
||
|
|
dataName = ncbi
|
||
|
|
# dataName = bc5cdr-disease
|
||
|
|
# dataName = bc5cdr-chemical
|
||
|
|
# dataName = bc2gm
|
||
|
|
|
||
|
|
[train]
|
||
|
|
epochs = 2
|
||
|
|
batch_size = 40
|
||
|
|
lr = 1e-5
|
||
|
|
|
||
|
|
dictionary = test_dictionary.txt
|
||
|
|
# or processed_dev_oneFile Note: bc2gm has no dev split
|
||
|
|
test_set = processed_test_refined
|
||
|
|
ckt_path = ./checkpoint/
|
||
|
|
|
||
|
|
|
||
|
|
[model]
|
||
|
|
margin = 2
|
||
|
|
model_name = dmis-lab/biobert-v1.1
|
||
|
|
|
||
|
|
[eval]
|
||
|
|
batch_size = 200
|
||
|
|
|
||
|
|
[data]
|
||
|
|
group_size = 4
|
||
|
|
shuffle = True
|