-
Notifications
You must be signed in to change notification settings - Fork 3
/
allen_conf_test_prep.json
50 lines (50 loc) · 1.11 KB
/
allen_conf_test_prep.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
45
46
47
48
49
50
{
"dataset_reader": {
"type": "mention_categories",
"category_mapping_file": "./data/test_category_mapping.json",
"tokenizer": {
"type": "word",
"word_splitter": {
"type": "fast-splitter"
}
},
"token_indexers": {
"tokens": {
"type": "fasttext-id",
"model_path": "./data/fasttext_embedding.model",
"model_params_path": "./data/fasttext_embedding.model.params"
}
}
},
"train_data_path": "./data/train_data_aa.tsv",
"validation_data_path": "data/train_data_aa.tsv",
"evaluate_on_test": true,
"model": {
"type": "fake_category_predictor"
},
"iterator": {
"type": "numpy-multiprocessing",
"num_workers": 1,
"base_iterator": {
"type": "basic",
"batch_size": 32
}
},
"trainer": {
"optimizer": {
"type": "adam",
"lr": 0.01
},
"should_log_learning_rate": true,
"num_epochs": 5,
"learning_rate_scheduler": {
"type": "reduce_on_plateau",
"factor": 0.5,
"patience": 5,
"verbose": true
}
},
"vocabulary": {
"directory_path": "./data/vocab2/"
}
}