def get_voca(self): try: # vocabulary 불러오기 ko_voc, en_voc = create_or_get_voca(save_path=self.args.dictionary_path) except OSError: # 경로 error 발생 시 각각의 경로를 입력해서 가지고 오기 ko_voc, en_voc = create_or_get_voca(save_path=self.args.dictionary_path, ko_corpus_path=self.x_train_path, di_corpus_path=self.y_train_path) return ko_voc, en_voc
def __init__(self, checkpoint, dictionary_path, x_path=None, y_path=None, beam_search=False, k=1): self.checkpoint = torch.load(checkpoint) self.seq_len = self.checkpoint['seq_len'] self.batch_size = 100 self.x_path = x_path self.y_path = y_path self.beam_search = beam_search self.k = k self.ko_voc, self.en_voc = create_or_get_voca( save_path=dictionary_path) self.model = self.model_load()
def __init__(self, checkpoint, dictionary_path, x_test_path=None, batch_size=1, beam_search=False, k=3, y_test_path=None, file_name=None): self.checkpoint = torch.load(checkpoint) self.seq_len = self.checkpoint['seq_len'] self.ko_voc, self.en_voc = create_or_get_voca( save_path=dictionary_path) self.beam_search = beam_search self.batch_size = batch_size if beam_search: self.beam = None self.k = k else: self.greedy = None