def METHOD_NAME(self, label, hidden, device):
    if label == self._SOS:
        return self._model.predict(None, hidden, add_sos=False)
    if label > self._blank_id:
        label -= 1
    label = label_collate([[label]]).to(device)
    return self._model.predict(label, hidden, add_sos=False)