LightChen2333
/

stack-propagation-slu-snips

Inference Endpoints

Model card Files Files and versions Community

LightChen2333 commited on Feb 17, 2023

Commit

3f78781

•

1 Parent(s): e03f735

Upload 3 files

Files changed (1) hide show

config.json +6 -3

config.json CHANGED Viewed

@@ -201,7 +201,9 @@
       "interaction": {
         "_model_target_": "model.decoder.interaction.StackInteraction",
         "differentiable": false,
-        "input_dim": 384
       },
       "slot_classifier": {
         "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",
@@ -211,7 +213,7 @@
         "force_ratio": 0.9,
         "hidden_dim": 64,
         "ignore_index": -100,
-        "input_dim": 384,
         "layer_num": 1,
         "mode": "slot",
         "return_sentence_level": false,
@@ -229,7 +231,8 @@
       },
       "embedding": {
         "dropout_rate": 0.4,
-        "embedding_dim": 256
       },
       "encoder_name": "self-attention-lstm",
       "lstm": {

       "interaction": {
         "_model_target_": "model.decoder.interaction.StackInteraction",
         "differentiable": false,
+        "input_dim": 384,
+        "intent_label_num": 7,
+        "slot_label_num": 72
       },
       "slot_classifier": {
         "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",
         "force_ratio": 0.9,
         "hidden_dim": 64,
         "ignore_index": -100,
+        "input_dim": 391,
         "layer_num": 1,
         "mode": "slot",
         "return_sentence_level": false,
       },
       "embedding": {
         "dropout_rate": 0.4,
+        "embedding_dim": 256,
+        "vocab_size": 11767
       },
       "encoder_name": "self-attention-lstm",
       "lstm": {