def save(self, filename):
     with ModelFile(filename, 'w') as fp:
         self.__vocab.save(fp.get_file_pointer())
         fp.write(self.__n_context)
         fp.write(self.__n_hidden)
         wrapper.begin_model_access(self.__model)
         fp.write_embed(self.__model.w_xh)
         fp.write_linear(self.__model.w_hy)
         wrapper.end_model_access(self.__model)
예제 #2
0
 def save(self, filename):
     with ModelFile(filename, 'w') as fp:
         self.__vocab.save(fp.get_file_pointer())
         fp.write(self.__n_context)
         fp.write(self.__n_hidden)
         wrapper.begin_model_access(self.__model)
         fp.write_embed(self.__model.w_xh)
         fp.write_linear(self.__model.w_hy)
         wrapper.end_model_access(self.__model)
예제 #3
0
 def load(filename):
     self = TransSegmentationModel()
     with ModelFile(filename) as fp:
         self.__vocab = Vocabulary.load(fp.get_file_pointer())
         self.__n_context = int(fp.read())
         self.__n_hidden = int(fp.read())
         self.__make_model()
         wrapper.begin_model_access(self.__model)
         fp.read_embed(self.__model.w_xh)
         fp.read_linear(self.__model.w_hy)
         wrapper.end_model_access(self.__model)
     return self
 def load(filename):
     self = SegmentationModel()
     with ModelFile(filename) as fp:
         self.__vocab = Vocabulary.load(fp.get_file_pointer())
         self.__n_context = int(fp.read())
         self.__n_hidden = int(fp.read())
         self.__make_model()
         wrapper.begin_model_access(self.__model)
         fp.read_embed(self.__model.w_xh)
         fp.read_linear(self.__model.w_hy)
         wrapper.end_model_access(self.__model)
     return self
 def save(self, filename):
     with ModelFile(filename, 'w') as fp:
         self.__vocab.save(fp.get_file_pointer())
         fp.write(self.__n_embed)
         fp.write(self.__n_hidden)
         wrapper.begin_model_access(self.__model)
         fp.write_embed(self.__model.w_xe)
         fp.write_linear(self.__model.w_ea)
         fp.write_linear(self.__model.w_aa)
         fp.write_linear(self.__model.w_eb)
         fp.write_linear(self.__model.w_bb)
         fp.write_linear(self.__model.w_ay1)
         fp.write_linear(self.__model.w_by1)
         fp.write_linear(self.__model.w_ay2)
         fp.write_linear(self.__model.w_by2)
         wrapper.end_model_access(self.__model)
 def save(self, filename):
     with ModelFile(filename, 'w') as fp:
         self.src_vocab.save(fp.get_file_pointer())
         self.trg_vocab.save(fp.get_file_pointer())
         fp.write(self.n_embed)
         fp.write(self.n_hidden)
         wrapper.begin_model_access(self.model)
         fp.write_embed(self.model.weight_xi)
         fp.write_linear(self.model.weight_ip)
         fp.write_linear(self.model.weight_pp)
         fp.write_linear(self.model.weight_pq)
         fp.write_linear(self.model.weight_qj)
         fp.write_linear(self.model.weight_jy)
         fp.write_embed(self.model.weight_yq)
         fp.write_linear(self.model.weight_qq)
         wrapper.end_model_access(self.model)
 def save(self, filename):
     with ModelFile(filename, 'w') as fp:
         self.src_vocab.save(fp.get_file_pointer())
         self.trg_vocab.save(fp.get_file_pointer())
         fp.write(self.n_embed)
         fp.write(self.n_hidden)
         wrapper.begin_model_access(self.model)
         fp.write_embed(self.model.weight_xi)
         fp.write_linear(self.model.weight_ip)
         fp.write_linear(self.model.weight_pp)
         fp.write_linear(self.model.weight_pq)
         fp.write_linear(self.model.weight_qj)
         fp.write_linear(self.model.weight_jy)
         fp.write_embed(self.model.weight_yq)
         fp.write_linear(self.model.weight_qq)
         wrapper.end_model_access(self.model)
 def load(self, filename):
     with ModelFile(filename) as fp:
         self.src_vocab = Vocabulary.load(fp.get_file_pointer())
         self.trg_vocab = Vocabulary.load(fp.get_file_pointer())
         self.n_embed = int(fp.read())
         self.n_hidden = int(fp.read())
         self.make_model()
         wrapper.begin_model_access(self.model)
         fp.read_embed(self.model.weight_xi)
         fp.read_linear(self.model.weight_ip)
         fp.read_linear(self.model.weight_pp)
         fp.read_linear(self.model.weight_pq)
         fp.read_linear(self.model.weight_qj)
         fp.read_linear(self.model.weight_jy)
         fp.read_embed(self.model.weight_yq)
         fp.read_linear(self.model.weight_qq)
         wrapper.end_model_access(self.model)
     return self
 def load(self, filename):
     with ModelFile(filename) as fp:
         self.src_vocab = Vocabulary.load(fp.get_file_pointer())
         self.trg_vocab = Vocabulary.load(fp.get_file_pointer())
         self.n_embed = int(fp.read())
         self.n_hidden = int(fp.read())
         self.make_model()
         wrapper.begin_model_access(self.model)
         fp.read_embed(self.model.weight_xi)
         fp.read_linear(self.model.weight_ip)
         fp.read_linear(self.model.weight_pp)
         fp.read_linear(self.model.weight_pq)
         fp.read_linear(self.model.weight_qj)
         fp.read_linear(self.model.weight_jy)
         fp.read_embed(self.model.weight_yq)
         fp.read_linear(self.model.weight_qq)
         wrapper.end_model_access(self.model)
     return self
 def load(filename):
     self = RNNSegmentationModel()
     with ModelFile(filename) as fp:
         self.__vocab = Vocabulary.load(fp.get_file_pointer())
         self.__n_embed = int(fp.read())
         self.__n_hidden = int(fp.read())
         self.__make_model()
         wrapper.begin_model_access(self.__model)
         fp.read_embed(self.__model.w_xe)
         fp.read_linear(self.__model.w_ea)
         fp.read_linear(self.__model.w_aa)
         fp.read_linear(self.__model.w_eb)
         fp.read_linear(self.__model.w_bb)
         fp.read_linear(self.__model.w_ay1)
         fp.read_linear(self.__model.w_by1)
         fp.read_linear(self.__model.w_ay2)
         fp.read_linear(self.__model.w_by2)
         wrapper.end_model_access(self.__model)
     return self