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
