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)
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 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