def _AddBeamReader(self,
task_context,
batch_size,
corpus_name,
until_all_final=False,
always_start_new_sentences=False):
"""Adds an op capable of reading sentences and parsing them with a beam."""
features, state, epochs = gen_parser_ops.beam_parse_reader(
task_context=task_context,
feature_size=self._feature_size,
beam_size=self._beam_size,
batch_size=batch_size,
corpus_name=corpus_name,
allow_feature_weights=self._allow_feature_weights,
arg_prefix=self._arg_prefix,
continue_until_all_final=until_all_final,
always_start_new_sentences=always_start_new_sentences)
return {'state': state, 'features': features, 'epochs': epochs}
def _AddBeamReader(self,
task_context,
batch_size,
corpus_name,
until_all_final=False,
always_start_new_sentences=False):
"""Adds an op capable of reading sentences and parsing them with a beam."""
features, state, epochs = gen_parser_ops.beam_parse_reader(
task_context=task_context,
feature_size=self._feature_size,
beam_size=self._beam_size,
batch_size=batch_size,
corpus_name=corpus_name,
allow_feature_weights=self._allow_feature_weights,
arg_prefix=self._arg_prefix,
continue_until_all_final=until_all_final,
always_start_new_sentences=always_start_new_sentences)
return {'state': state, 'features': features, 'epochs': epochs}