def __init__(self, opt, shared=None):
build(opt)
opt['datafile'] = os.path.join(
opt['datapath'],
'dialog-bAbI-plus',
'dialog-bAbI-plus-tasks',
'dialog-babi-kb-all.txt',
)
super().__init__(opt, shared)
def _path(task, opt):
# Build the data if it doesn't exist.
build(opt)
prefix = os.path.join(opt['datapath'], 'dialog-bAbI-plus', 'dialog-bAbI-plus-tasks')
suffix = ''
dt = opt['datatype'].split(':')[0]
if dt == 'train':
suffix = 'trn'
elif dt == 'test':
suffix = 'tst'
elif dt == 'valid':
suffix = 'dev'
datafile = os.path.join(prefix,
'{tsk}-{type}.txt'.format(tsk=tasks[int(task)], type=suffix))
cands_datafile = os.path.join(prefix, 'dialog-babi-candidates.txt')
return datafile, cands_datafile
def _path(task, opt):
# Build the data if it doesn't exist.
build(opt)
prefix = os.path.join(opt['datapath'], 'dialog-bAbI-plus', 'dialog-bAbI-plus-tasks')
suffix = ''
dt = opt['datatype'].split(':')[0]
if dt == 'train':
suffix = 'trn'
elif dt == 'test':
suffix = 'tst'
elif dt == 'valid':
suffix = 'dev'
datafile = os.path.join(prefix,
'{tsk}-{type}.txt'.format(tsk=tasks[int(task)], type=suffix))
cands_datafile = os.path.join(prefix, 'dialog-babi-candidates.txt')
return datafile, cands_datafile
def __init__(self, opt, shared=None):
build(opt)
opt['datafile'] = os.path.join(opt['datapath'], 'dialog-bAbI-plus',
'dialog-bAbI-plus-tasks',
'dialog-babi-kb-all.txt')
super().__init__(opt, shared)