Example #1
0
 def load_input_embeddings_path(self, data):
     self.input_embeddings_path = get_value_as_str_list(
         data.get("input_embeddings_path", None))
     self.input_embeddings_path = [
         get_real_path(path, self.config_root_path)
         for path in self.input_embeddings_path
     ]
     pad_list(self.input_embeddings_path, self.input_features, None)
Example #2
0
 def load_input_embeddings_sizes(self, data):
     self.input_embeddings_sizes = get_value_as_int_list(
         data.get("input_embeddings_size", None))
     pad_list(self.input_embeddings_sizes, self.data_config.input_features,
              300)
     self.reset_input_embeddings_sizes()
Example #3
0
 def set_input_resize(self, input_resize):
     self.input_resize = get_value_as_int_list(input_resize)
     pad_list(self.input_resize, self.data_config.input_features, None)
     for i in range(len(self.input_resize)):
         if self.input_resize[i] is not None and self.input_resize[i] <= 0:
             self.input_resize[i] = None
Example #4
0
 def set_input_auto_model(self, auto_model, auto_path):
     self.input_auto_model = get_value_as_str_list(auto_model)
     self.input_auto_path = get_value_as_str_list(auto_path)
     pad_list(self.input_auto_model, self.data_config.input_features, None)
     pad_list(self.input_auto_path, self.data_config.input_features, None)
     self.reset_input_embeddings_sizes()
Example #5
0
 def set_input_bert_model(self, bert_model):
     self.input_bert_path = get_value_as_str_list(bert_model)
     pad_list(self.input_bert_path, self.data_config.input_features, None)
     self.reset_input_embeddings_sizes()
Example #6
0
 def set_input_elmo_path(self, elmo_path):
     self.input_elmo_path = get_value_as_str_list(elmo_path)
     pad_list(self.input_elmo_path, self.data_config.input_features, None)
     self.reset_input_embeddings_sizes()
Example #7
0
 def set_input_embeddings_tokenize_model(self, tokenize_model):
     self.input_embeddings_tokenize_model = get_value_as_str_list(
         tokenize_model)
     pad_list(self.input_embeddings_tokenize_model,
              self.data_config.input_features, None)
Example #8
0
 def load_output_translation_clear_text_values(self, data):
     self.output_translation_clear_text = get_value_as_bool_list(
         data.get("output_translation_clear_text", None))
     pad_list(self.output_translation_clear_text,
              self.output_translation_features, False)
Example #9
0
 def load_input_clear_text_values(self, data):
     self.input_clear_text = get_value_as_bool_list(
         data.get("input_clear_text", None))
     pad_list(self.input_clear_text, self.input_features, False)