|
|
import os |
|
|
import torch |
|
|
from torch import nn |
|
|
|
|
|
from common import utils |
|
|
from common.utils import ClassifierOutputData, HiddenData |
|
|
from model.decoder.interaction.base_interaction import BaseInteraction |
|
|
|
|
|
|
|
|
class StackInteraction(BaseInteraction): |
|
|
def __init__(self, **config): |
|
|
super().__init__(**config) |
|
|
self.intent_embedding = nn.Embedding( |
|
|
self.config["intent_label_num"], self.config["intent_label_num"] |
|
|
) |
|
|
self.differentiable = config.get("differentiable") |
|
|
self.intent_embedding.weight.data = torch.eye( |
|
|
self.config["intent_label_num"]) |
|
|
self.intent_embedding.weight.requires_grad = False |
|
|
|
|
|
def forward(self, intent_output: ClassifierOutputData, encode_hidden: HiddenData): |
|
|
if not self.differentiable: |
|
|
_, idx_intent = intent_output.classifier_output.topk(1, dim=-1) |
|
|
feed_intent = self.intent_embedding(idx_intent.squeeze(2)) |
|
|
else: |
|
|
feed_intent = intent_output.classifier_output |
|
|
encode_hidden.update_slot_hidden_state( |
|
|
torch.cat([encode_hidden.get_slot_hidden_state(), feed_intent], dim=-1)) |
|
|
return encode_hidden |
|
|
|
|
|
@staticmethod |
|
|
def from_configured(configure_name_or_file="stack-interaction", **input_config): |
|
|
return utils.from_configured(configure_name_or_file, |
|
|
model_class=StackInteraction, |
|
|
config_prefix="./config/decoder/interaction", |
|
|
**input_config) |
|
|
|
