def output_ports(self): """Returns definitions of module output ports. """ return { "punct_logits": NeuralType(('B', 'T', 'D'), LogitsType()), "capit_logits": NeuralType(('B', 'T', 'D'), LogitsType()), }
def output_ports(self): """Returns definitions of module output ports. intent_logits: TODO slot_logits: TODO """ return { "intent_logits": NeuralType(('B', 'D'), LogitsType()), "slot_logits": NeuralType(('B', 'T', 'D'), LogitsType()), }
def input_ports(self): """Returns definitions of module input ports. """ return { # "intent_logits": NeuralType({0: AxisType(BatchTag), 1: AxisType(ChannelTag)}), # "slot_logits": NeuralType({0: AxisType(BatchTag), 1: AxisType(TimeTag), 2: AxisType(ChannelTag)}), # "loss_mask": NeuralType({0: AxisType(BatchTag), 1: AxisType(TimeTag)}), # "intents": NeuralType({0: AxisType(BatchTag)}), # "slots": NeuralType({0: AxisType(BatchTag), 1: AxisType(TimeTag)}), "intent_logits": NeuralType(('B', 'D'), LogitsType()), "slot_logits": NeuralType(('B', 'T', 'D'), LogitsType()), "loss_mask": NeuralType(('B', 'T'), ChannelType()), "intents": NeuralType(tuple('B'), ChannelType()), "slots": NeuralType(('B', 'T'), ChannelType()), }
def input_ports(self): """Returns definitions of module input ports. """ return { "logits": NeuralType(('B', 'T', 'D'), LogitsType()), "labels": NeuralType(('B', 'T'), LabelsType()), "output_mask": NeuralType(('B', 'T'), MaskType(), optional=True), }
def input_ports(self): """Returns definitions of module input ports. """ return { "logits": NeuralType(('B', 'T', 'D'), LogitsType()), "start_positions": NeuralType(tuple('B'), ChannelType()), "end_positions": NeuralType(tuple('B'), ChannelType()), }
def input_ports(self): """Returns definitions of module input ports. """ return { # "logits": NeuralType({0: AxisType(BatchTag), 1: AxisType(TimeTag), 2: AxisType(ChannelTag)}), # "target_ids": NeuralType({0: AxisType(BatchTag), 1: AxisType(TimeTag)}), "logits": NeuralType(('B', 'T', 'D'), LogitsType()), "target_ids": NeuralType(('B', 'T'), LabelsType()), }
def input_ports(self): """Returns definitions of module input ports. """ return { # "logits": NeuralType({0: AxisType(BatchTag), 1: AxisType(TimeTag), 2: AxisType(ChannelTag)}), # "start_positions": NeuralType({0: AxisType(BatchTag)}), # "end_positions": NeuralType({0: AxisType(BatchTag)}), "logits": NeuralType(('B', 'T', 'D'), LogitsType()), "start_positions": NeuralType(tuple('B'), ChannelType()), "end_positions": NeuralType(tuple('B'), ChannelType()), }
def output_ports(self): """Returns definitions of module output ports. intent_logits: 0: AxisType(BatchTag) 1: AxisType(ChannelTag) slot_logits: 0: AxisType(BatchTag) 1: AxisType(TimeTag) 2: AxisType(ChannelTag) """ return { # "intent_logits": NeuralType({0: AxisType(BatchTag), 1: AxisType(ChannelTag)}), # "slot_logits": NeuralType({0: AxisType(BatchTag), 1: AxisType(TimeTag), 2: AxisType(ChannelTag)}), "intent_logits": NeuralType(('B', 'D'), LogitsType()), "slot_logits": NeuralType(('B', 'T', 'D'), LogitsType()), }
def input_ports(self): """Returns definitions of module input ports. logit_intent_status (float): Output of SGD model intent_status_labels (int): Intent labels logit_req_slot_status (float): Output of SGD model requested_slot_status (float): Takes value 1 if the corresponding slot is requested, 0 otherwise req_slot_mask (bool): Masks requested slots not used for the particular service logit_cat_slot_status (float): Output of SGD model categorical_slot_status (int): The status of each categorical slot in the service cat_slot_status_mask (bool): Masks categorical slots not used for the particular service logit_cat_slot_value (float): Output of SGD model categorical_slot_values (int): The index of the correct value for each categorical slot logit_noncat_slot_status (float): Output of SGD model noncategorical_slot_status (int): The status of each noncategorical slot in the service noncat_slot_status_mask (bool): masks noncategorical slots not used for the particular service logit_noncat_slot_start (float): Output of SGD model logit_noncat_slot_end (float): Output of SGD model noncategorical_slot_value_start (int): The index of the starting subword corresponding to the slot span for a non-categorical slot value noncategorical_slot_value_end (int): The index of the ending (inclusive) subword corresponding to the slot span for a non-categorical slot value """ return { "logit_intent_status": NeuralType(('B', 'T', 'C'), LogitsType()), "intent_status_labels": NeuralType(('B'), LabelsType()), "logit_req_slot_status": NeuralType(('B', 'T'), LogitsType()), "requested_slot_status": NeuralType(('B', 'T'), LabelsType()), "req_slot_mask": NeuralType(('B', 'T'), ChannelType()), "logit_cat_slot_status": NeuralType(('B', 'T', 'C'), LogitsType()), "categorical_slot_status": NeuralType(('B', 'T'), LabelsType()), "cat_slot_status_mask": NeuralType(('B', 'T'), ChannelType()), "logit_cat_slot_value": NeuralType(('B', 'T', 'C'), LogitsType()), "categorical_slot_values": NeuralType(('B', 'T'), LabelsType()), "logit_noncat_slot_status": NeuralType(('B', 'T', 'C'), LogitsType()), "noncategorical_slot_status": NeuralType(('B', 'T'), LabelsType()), "noncat_slot_status_mask": NeuralType(('B', 'T'), ChannelType()), "logit_noncat_slot_start": NeuralType(('B', 'T', 'C'), LogitsType()), "logit_noncat_slot_end": NeuralType(('B', 'T', 'C'), LogitsType()), "noncategorical_slot_value_start": NeuralType(('B', 'T'), LabelsType()), "noncategorical_slot_value_end": NeuralType(('B', 'T'), LabelsType()), }
def output_ports(self): """Returns definitions of module output ports. logit_intent_status (float): output for intent status logit_req_slot_status (float): output for requested slots status logit_cat_slot_status (float): output for categorical slots status logit_cat_slot_value (float): output for categorical slots values logit_noncat_slot_status (float): Output of SGD model logit_noncat_slot_start (float): output for non categorical slots values start logit_noncat_slot_end (float): output for non categorical slots values end """ return { "logit_intent_status": NeuralType(('B', 'T', 'C'), LogitsType()), "logit_req_slot_status": NeuralType(('B', 'T'), LogitsType()), "logit_cat_slot_status": NeuralType(('B', 'T', 'C'), LogitsType()), "logit_cat_slot_value": NeuralType(('B', 'T', 'C'), LogitsType()), "logit_noncat_slot_status": NeuralType(('B', 'T', 'C'), LogitsType()), "logit_noncat_slot_start": NeuralType(('B', 'T', 'C'), LogitsType()), "logit_noncat_slot_end": NeuralType(('B', 'T', 'C'), LogitsType()), }
def output_ports(self): """Returns definitions of module output ports. logits: logits before loss """ return {"logits": NeuralType(('B', 'T', 'C'), LogitsType())}
def output_ports(self): """Returns definitions of module output ports. """ # return {"logits": NeuralType({0: AxisType(BatchTag), 1: AxisType(ChannelTag)})} return {"logits": NeuralType(('B', 'D'), LogitsType())}