'Extends the event sequences by sampling the softmax probabilities.
Args:
target_event_sequences: A list of target EventSequence objects.
softmax: A list of softmax probability vectors. The list of softmaxes
should be the same length as the list of event sequences.
Returns:
A Python list of chosen class indices, one for each target event sequence.'

def extend_event_sequences(self, target_event_sequences, softmax):

    return self._target_encoder_decoder.extend_event_sequences(target_event_sequences, softmax)
