LabelClassifier¶
-
class
LabelClassifier(input_size, classifier_hidden_size, classes_label=39, bias=True, dropout=0.5)[source]¶ Label Classifier model to be used in RST parser network model.
-
forward(input_left: torch.Tensor, input_right: torch.Tensor) → Tuple[torch.Tensor, torch.Tensor][source]¶ Forward pass for the label classifier.
- Parameters
input_left (torch.Tensor) – encoder RNN output
input_right (torch.Tensor) – encoder RNN output
- Returns
return softmax and log softmax tensors of label classifier model.
- Return type
Tuple[torch.Tensor, torch.Tensor]
-