deep_gru_lstm_save.py

def save(self, name: str) -> None:
    mkdir(f'./{name}')
    mkdir(f'./{name}/weights')
    with open(f'./{name}/vocabulary.txt', 'w') as f:
        f.write('[separator]'.join(self.vocab))
    with open(f'./{name}/inter_time_step_size.txt', 'w') as f:
        f.write(str(self.inter_time_step_size))
    with open(f'./{name}/unit_type.txt', 'w') as f:
        f.write(self.unit_type)
    with open(f'./{name}/depth.txt', 'w') as f:
        f.write(str(self.depth))
    
    if self.unit_type == 'gru':
        for s in ['wr', 'br', 'wu', 'bu', 'wa', 'ba']:
            for i in range(self.depth):
                np.save(f'./{name}/weights/{s}_{i}.npy',
                        getattr(self, s)[i].numpy())
    else:
        for s in ['wu', 'bu', 'wf', 'bf', 'wo', 'bo', 'wc', 'bc']:
            for i in range(self.depth):
                np.save(f'./{name}/weights/{s}_{i}.npy',
                        getattr(self, s)[i].numpy())
    
    np.save(f'./{name}/weights/wy.npy', self.wy.numpy())
    np.save(f'./{name}/weights/by.npy', self.by.numpy())