ConvLab
/

sumbt-dst-multiwoz21

+{
+  "_name_or_path": "ConvLab/sumbt-dst-multiwoz21",
+  "architectures": [
+    "RobertaSetSUMBT"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "bos_token_id": 0,
+  "candidate_embedding_model_name": "roberta-base",
+  "candidate_pooling": "mean",
+  "classifier_dropout": null,
+  "dev_batch_size": 16,
+  "distance_measure": "cosine",
+  "domain_ids": {
+    "attraction": [
+      1,
+      2,
+      3,
+      4,
+      5,
+      6,
+      7,
+      8,
+      9
+    ],
+    "hospital": [
+      10,
+      11,
+      12,
+      13
+    ],
+    "hotel": [
+      14,
+      15,
+      16,
+      17,
+      18,
+      19,
+      20,
+      21,
+      22,
+      23,
+      24,
+      25,
+      26,
+      27,
+      28
+    ],
+    "police": [
+      58,
+      59,
+      60,
+      61
+    ],
+    "restaurant": [
+      32,
+      33,
+      34,
+      35,
+      36,
+      37,
+      38,
+      39,
+      40,
+      29,
+      30,
+      31
+    ],
+    "taxi": [
+      41,
+      42,
+      43,
+      44,
+      45,
+      46
+    ],
+    "train": [
+      47,
+      48,
+      49,
+      50,
+      51,
+      52,
+      53,
+      54,
+      55,
+      56,
+      57
+    ]
+  },
+  "dropout_rate": 0.3,
+  "eos_token_id": 2,
+  "freeze_encoder": false,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "informable_slot_ids": {
+    "attraction-area": 2,
+    "attraction-name": 5,
+    "attraction-type": 9,
+    "hospital-department": 11,
+    "hotel-area": 15,
+    "hotel-book day": 16,
+    "hotel-book people": 17,
+    "hotel-book stay": 18,
+    "hotel-internet": 20,
+    "hotel-name": 21,
+    "hotel-parking": 22,
+    "hotel-price range": 25,
+    "hotel-stars": 27,
+    "hotel-type": 28,
+    "restaurant-area": 30,
+    "restaurant-book day": 31,
+    "restaurant-book people": 32,
+    "restaurant-book time": 33,
+    "restaurant-food": 35,
+    "restaurant-name": 36,
+    "restaurant-price range": 39,
+    "taxi-arrive by": 41,
+    "taxi-departure": 42,
+    "taxi-destination": 43,
+    "taxi-leave at": 44,
+    "train-arrive by": 47,
+    "train-book people": 48,
+    "train-day": 50,
+    "train-departure": 51,
+    "train-destination": 52,
+    "train-leave at": 54
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label_smoothing": 0.05,
+  "layer_norm_eps": 1e-05,
+  "loss_function": "labelsmoothing",
+  "max_candidate_len": 12,
+  "max_dialogue_len": 12,
+  "max_position_embeddings": 514,
+  "max_slot_len": 12,
+  "max_turn_len": 64,
+  "model_type": "roberta",
+  "nbt_hidden_size": 300,
+  "nbt_layers": 1,
+  "nbt_type": "gru",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "num_slots": 62,
+  "num_values": {
+    "attraction-address": 1,
+    "attraction-area": 7,
+    "attraction-choice": 1,
+    "attraction-entrance fee": 1,
+    "attraction-name": 156,
+    "attraction-open hours": 1,
+    "attraction-phone": 1,
+    "attraction-postcode": 1,
+    "attraction-type": 14,
+    "hospital-address": 1,
+    "hospital-department": 49,
+    "hospital-phone": 1,
+    "hospital-postcode": 1,
+    "hotel-address": 1,
+    "hotel-area": 7,
+    "hotel-book day": 9,
+    "hotel-book people": 12,
+    "hotel-book stay": 12,
+    "hotel-choice": 1,
+    "hotel-internet": 5,
+    "hotel-name": 72,
+    "hotel-parking": 5,
+    "hotel-phone": 1,
+    "hotel-postcode": 1,
+    "hotel-price range": 5,
+    "hotel-ref": 1,
+    "hotel-stars": 8,
+    "hotel-type": 5,
+    "police-address": 1,
+    "police-name": 1,
+    "police-phone": 1,
+    "police-postcode": 1,
+    "restaurant-address": 1,
+    "restaurant-area": 7,
+    "restaurant-book day": 9,
+    "restaurant-book people": 12,
+    "restaurant-book time": 290,
+    "restaurant-choice": 1,
+    "restaurant-food": 103,
+    "restaurant-name": 185,
+    "restaurant-phone": 1,
+    "restaurant-postcode": 1,
+    "restaurant-price range": 5,
+    "restaurant-ref": 1,
+    "taxi-arrive by": 290,
+    "taxi-departure": 255,
+    "taxi-destination": 253,
+    "taxi-leave at": 290,
+    "taxi-phone": 1,
+    "taxi-type": 1,
+    "train-arrive by": 290,
+    "train-book people": 12,
+    "train-choice": 1,
+    "train-day": 9,
+    "train-departure": 32,
+    "train-destination": 28,
+    "train-duration": 1,
+    "train-leave at": 290,
+    "train-price": 1,
+    "train-ref": 1,
+    "train-train id": 1
+  },
+  "pad_token_id": 1,
+  "position_embedding_type": "absolute",
+  "predict_actions": false,
+  "requestable_slot_ids": {
+    "attraction-address": 1,
+    "attraction-area": 2,
+    "attraction-entrance fee": 4,
+    "attraction-phone": 7,
+    "attraction-postcode": 8,
+    "attraction-type": 9,
+    "hospital-address": 10,
+    "hospital-phone": 12,
+    "hospital-postcode": 13,
+    "hotel-address": 14,
+    "hotel-area": 15,
+    "hotel-internet": 20,
+    "hotel-parking": 22,
+    "hotel-phone": 23,
+    "hotel-postcode": 24,
+    "hotel-price range": 25,
+    "hotel-ref": 26,
+    "hotel-stars": 27,
+    "hotel-type": 28,
+    "police-address": 58,
+    "police-phone": 60,
+    "police-postcode": 61,
+    "restaurant-address": 29,
+    "restaurant-area": 30,
+    "restaurant-food": 35,
+    "restaurant-phone": 37,
+    "restaurant-postcode": 38,
+    "restaurant-price range": 39,
+    "restaurant-ref": 40,
+    "taxi-phone": 45,
+    "taxi-type": 46,
+    "train-arrive by": 47,
+    "train-duration": 53,
+    "train-leave at": 54,
+    "train-price": 55,
+    "train-ref": 56,
+    "train-train id": 57
+  },
+  "rnn_zero_init": false,
+  "seed": 0,
+  "set_similarity": false,
+  "slot_attention_heads": 12,
+  "slot_ids": {
+    "attraction-address": 1,
+    "attraction-area": 2,
+    "attraction-choice": 3,
+    "attraction-entrance fee": 4,
+    "attraction-name": 5,
+    "attraction-open hours": 6,
+    "attraction-phone": 7,
+    "attraction-postcode": 8,
+    "attraction-type": 9,
+    "hospital-address": 10,
+    "hospital-department": 11,
+    "hospital-phone": 12,
+    "hospital-postcode": 13,
+    "hotel-address": 14,
+    "hotel-area": 15,
+    "hotel-book day": 16,
+    "hotel-book people": 17,
+    "hotel-book stay": 18,
+    "hotel-choice": 19,
+    "hotel-internet": 20,
+    "hotel-name": 21,
+    "hotel-parking": 22,
+    "hotel-phone": 23,
+    "hotel-postcode": 24,
+    "hotel-price range": 25,
+    "hotel-ref": 26,
+    "hotel-stars": 27,
+    "hotel-type": 28,
+    "police-address": 58,
+    "police-name": 59,
+    "police-phone": 60,
+    "police-postcode": 61,
+    "restaurant-address": 29,
+    "restaurant-area": 30,
+    "restaurant-book day": 31,
+    "restaurant-book people": 32,
+    "restaurant-book time": 33,
+    "restaurant-choice": 34,
+    "restaurant-food": 35,
+    "restaurant-name": 36,
+    "restaurant-phone": 37,
+    "restaurant-postcode": 38,
+    "restaurant-price range": 39,
+    "restaurant-ref": 40,
+    "taxi-arrive by": 41,
+    "taxi-departure": 42,
+    "taxi-destination": 43,
+    "taxi-leave at": 44,
+    "taxi-phone": 45,
+    "taxi-type": 46,
+    "train-arrive by": 47,
+    "train-book people": 48,
+    "train-choice": 49,
+    "train-day": 50,
+    "train-departure": 51,
+    "train-destination": 52,
+    "train-duration": 53,
+    "train-leave at": 54,
+    "train-price": 55,
+    "train-ref": 56,
+    "train-train id": 57
+  },
+  "test_batch_size": 16,
+  "torch_dtype": "float32",
+  "train_batch_size": 3,
+  "transformers_version": "4.24.0",
+  "type_vocab_size": 1,
+  "use_cache": true,
+  "use_descriptions": true,
+  "vocab_size": 50265
+}