/
config.yml
55 lines (55 loc) · 1.1 KB
/
config.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
55
mode: term
base_path: ./data/MAMS-ATSA
glove_path: ./data/glove.840B.300d.txt
sentiment_path: ./data/sentiment_dict.json
lowercase: True
max_vocab_size: None
min_vocab_freq: 0
aspect_term_model:
type: recurrent_capsnet
recurrent_capsnet:
embed_size: 300
dropout: 0.5
num_layers: 2
capsule_size: 300
bidirectional: True
optimizer: adam
batch_size: 64
learning_rate: 0.0003
weight_decay: 0
num_epoches: 20
gpu: 0
bert_capsnet:
bert_size: 768
capsule_size: 300
dropout: 0.1
optimizer: adam
batch_size: 32
learning_rate: 0.00002
weight_decay: 0
num_epoches: 5
gpu: 0
aspect_category_model:
type: recurrent_capsnet
recurrent_capsnet:
embed_size: 300
dropout: 0.5
num_layers: 2
capsule_size: 300
bidirectional: True
optimizer: adam
batch_size: 64
learning_rate: 0.0003
weight_decay: 0
num_epoches: 20
gpu: 0
bert_capsnet:
bert_size: 768
capsule_size: 300
dropout: 0.1
optimizer: adam
batch_size: 32
learning_rate: 0.00003
weight_decay: 0
num_epoches: 5
gpu: 0