forked from shanybar/event_entity_coref_ecb_plus
-
Notifications
You must be signed in to change notification settings - Fork 1
/
train_config.json
44 lines (38 loc) · 1.21 KB
/
train_config.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
{
"batch_size": 16,
"char_pretrained_path": "data/external/char_embed/glove.840B.300d-char.npy",
"char_rep_size": 50,
"char_vocab_path": "data/external/char_embed/glove.840B.300d-char.vocab",
"dev_path": "out_features/ELMO/dev_data",
"dev_th_range": [0.5, 0.6],
"entity_merge_threshold": 0.5,
"epochs": 50,
"event_merge_threshold": 0.5,
"feature_size": 50,
"glove_path": "trained_model_n_data/glove.6B.300d.txt",
"ft_path": "trained_model_n_data/glove.6B.300d.txt.vec",
"gpu_num": 0,
"load_predicted_topics": false,
"log_interval": 32,
"loss": "bce",
"lr": 0.0001,
"merge_iters": 2,
"momentum": 0,
"optimizer": "adam",
"patient": 7,
"random_seed": 2048,
"regressor_epochs": 1,
"remove_singletons": false,
"resume_training": false,
"seed": 1,
"test_use_gold_mentions": true,
"train_init_wd_entity_with_gold": true,
"train_path": "out_features/ELMO/training_data",
"use_args_feats": true,
"use_binary_feats": true,
"use_diff": false,
"use_mult": true,
"use_pretrained_char": true,
"wd_entity_coref_file": "data/external/stanford_neural_wd_entity_coref_out/ecb_wd_coref.json",
"weight_decay": 0
}