forked from facebookresearch/pytext
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathseqnn.json
43 lines (43 loc) · 936 Bytes
/
seqnn.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
{
"version":7,
"task": {
"SeqNNTask": {
"data": {
"source": {
"TSVDataSource": {
"field_names": ["label", "text_seq"],
"train_filename": "tests/data/msg_topic_train.tsv",
"test_filename": "tests/data/msg_topic_train.tsv",
"eval_filename": "tests/data/msg_topic_train.tsv"
}
}
},
"trainer": {
"epochs": 2
},
"model": {
"representation": {
"doc_representation": {
"dropout": 0.5
},
"seq_representation": {
"BiLSTMDocAttention": {
"dropout": 0.5
}
}
},
"output_layer": {
"loss": {
"BinaryCrossEntropyLoss": {}
}
},
"decoder": {
"hidden_dims": [
100
]
}
}
}
},
"use_cuda_if_available": false
}