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