コード例 #1
0
 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}
コード例 #2
0
 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}