Method encode

openrec/preprocess/dptr_label_encode.py:113–118

(self,
               labels: List[str],
               device: Optional[torch.device] = None)

        ]

    def encode(self,
               labels: List[str],
               device: Optional[torch.device] = None) -> Tensor:
        batch = [self.bos_id] + self._tok2ids(labels) + [self.eos_id]
        return batch
        # return pad_sequence(batch, batch_first=True, padding_value=self.pad_id)

    def _filter(self, probs: Tensor, ids: Tensor) -> Tuple[Tensor, List[int]]:
        ids = ids.tolist()