-
Notifications
You must be signed in to change notification settings - Fork 0
/
xml_cnn_np2_best.yml
54 lines (54 loc) · 946 Bytes
/
xml_cnn_np2_best.yml
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
51
52
53
54
activation: relu
batch_size: 64
config: config/EUR-Lex/xml_cnn_np2_best.yml
cpu: false
data_dir: data/EUR-Lex
data_name: EUR-Lex
data_workers: 4
display_iter: 100
dropout: 0.2
dropout2: 0.4
embed_cache_dir: .vector_cache
embed_file: glove.6B.300d
epochs: 50
eval: false
eval_batch_size: 256
eval_last: true
filter_sizes:
- 8
fixed_length: false
hidden_dim: 1024
init_weight: kaiming_uniform
label_file: null
learning_rate: 0.0003
load_checkpoint: null
max_seq_length: 500
metrics_thresholds: 0.5
min_vocab_freq: 1
model_name: XMLCNN
momentum: 0.9
monitor_metrics:
- P@1
- P@3
- P@5
- nDCG@1
- nDCG@3
- nDCG@5
num_filter_per_size: 768
num_pool: 2
optimizer: adam
patience: 5
predict_out_path: null
result_dir: runs
search_alg: optuna
seed: null
shuffle: true
silent: false
test_path: data/EUR-Lex/test_rv.txt
train_path: data/EUR-Lex/train_rv.txt
use_extended_loss: null
val_metric: P@1
val_path: null
val_size: 0
vocab_file: null
weight_decay: 0