Skip to content
This repository was archived by the owner on Nov 22, 2022. It is now read-only.

Commit 62d9ec1

Browse files
TitousenseiEric Gaudet
authored andcommitted
ATIS tutorial rewrite
This is a partial solution with the following problems: - tensorboard needs to be installed to export a c2 model ??? - slot f1-score is 19%, looks like BIO mismatch
1 parent 8f93ce1 commit 62d9ec1

File tree

8 files changed

+460
-389
lines changed

8 files changed

+460
-389
lines changed

demo/atis_joint_model/atis_joint_config.json

Lines changed: 0 additions & 81 deletions
This file was deleted.

demo/atis_joint_model/data_processor.py

Lines changed: 0 additions & 199 deletions
This file was deleted.

demo/configs/atis_intent_slot.json

Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,52 @@
1+
{
2+
"include_dirs": ["demo/datasource"],
3+
"export_caffe2_path": "atis_intent_slot.c2",
4+
"task": {
5+
"IntentSlotTask": {
6+
"data": {
7+
"Data": {
8+
"source": {
9+
"AtisIntentSlotsDataSource": {
10+
"path": "atis"
11+
}
12+
},
13+
"batcher": {
14+
"PoolingBatcher": {
15+
"train_batch_size": 128,
16+
"eval_batch_size": 128,
17+
"test_batch_size": 128
18+
}
19+
},
20+
"sort_key": "tokens"
21+
}
22+
},
23+
"model": {
24+
"representation": {
25+
"BiLSTMDocSlotAttention": {
26+
"pooling": {
27+
"SelfAttention": {}
28+
}
29+
}
30+
},
31+
"output_layer": {
32+
"doc_output": {
33+
"loss": {
34+
"CrossEntropyLoss": {}
35+
}
36+
},
37+
"word_output": {
38+
"CRFOutputLayer": {}
39+
}
40+
},
41+
"word_embedding": {
42+
"embed_dim": 100,
43+
"pretrained_embeddings_path": "atis/glove.6B.100d.txt"
44+
}
45+
},
46+
"trainer": {
47+
"epochs": 20
48+
}
49+
}
50+
},
51+
"version": 17
52+
}

0 commit comments

Comments
 (0)