def __init__(self, opt): self.personas_idx_stack_path = os.path.join( opt['extract_personas_path'], './personas_idx_stack.pkl') self.personas_path = '{}/personas-{}'.format( opt['extract_personas_path'], opt['persona_type'] + 'Revised' if opt['revised'] else 'Original') if not os.path.exists(self.personas_path): opt['personas_path'] = self.personas_path main_extract(opt) self.personas_name_list = [] for f_name in os.listdir(self.personas_path): if f_name.endswith('.pkl'): self.personas_name_list.append(f_name) if os.path.exists(self.personas_idx_stack_path): with open(self.personas_idx_stack_path, 'rb') as handle: self.idx_stack = pickle.load(handle) else: self.idx_stack = [] self.add_idx_stack() self.save_idx_stack() pass
def __init__(self, opt): self.personas_idx_stack_path = os.path.join( os.getcwd(), './personas_idx_stack.pkl') self.personas_path = '{}/data/personas-{}'.format( os.getcwd(), opt['persona_type'] + 'Original') self.topic_for_personas_path = '{}/personas_with_wiki_links.txt'.format( os.getcwd()) if not os.path.exists(self.personas_path): opt['personas_path'] = self.personas_path main_extract(opt) self.personas_name_list = [] for f_name in os.listdir(self.personas_path): if f_name.endswith('.pkl'): self.personas_name_list.append(f_name) if os.path.exists(self.personas_idx_stack_path): with open(self.personas_idx_stack_path, 'rb') as handle: self.idx_stack = pickle.load(handle) else: self.idx_stack = [] self.add_idx_stack() self.save_idx_stack() self.load_topics_for_personas()