def convert2SQL(model, datas):
# loading dataSets
#datas, schemas = load_dataSets(args)
#datas, schemas = load_predict_dataset(args)
schemas = model.tables
print('Converting semantic to sql...')
#print(f'> Start: {datas}')
alter_not_in(datas, schemas=schemas)
alter_inter(datas)
alter_column0(datas)
#print(f'> After: {datas}')
result = transform(datas[0], schemas[datas[0]['db_id']])
return result[0]
arg_parser = argparse.ArgumentParser()
arg_parser.add_argument('--data_path',
type=str,
help='dataset path',
required=True)
arg_parser.add_argument('--input_path',
type=str,
help='predicted logical form',
required=True)
arg_parser.add_argument('--output_path', type=str, help='output data')
args = arg_parser.parse_args()
# loading dataSets
datas, schemas = load_dataSets(args)
alter_not_in(datas, schemas=schemas)
alter_inter(datas)
alter_column0(datas)
index = range(len(datas))
count = 0
exception_count = 0
with open(args.output_path,
'w', encoding='utf8') as d, open('gold.txt',
'w',
encoding='utf8') as g:
for i in index:
try:
result = transform(datas[i], schemas[datas[i]['db_id']])
d.write(result[0] + '\n')
g.write("%s\t%s\t%s\n" % (datas[i]['query'], datas[i]["db_id"],
datas[i]["question"]))
return sql
if __name__ == '__main__':
arg_parser = argparse.ArgumentParser()
arg_parser.add_argument('--data_path', default='../IRNet-dev/data', type=str, help='dataset path')
arg_parser.add_argument('--input_path', default='predict_lf.json', type=str, help='predicted logical form')
arg_parser.add_argument('--output_path', default='predict.sql', type=str, help='output data')
arg_parser.add_argument('--v2',default=False)
args = arg_parser.parse_args()
# loading dataSets
datas, schemas = load_dataSets(args)
alter_not_in(args,datas, schemas=schemas)
alter_inter(args,datas)
alter_column0(args,datas)
index = range(len(datas))
count = 0
exception_count = 0
with open(args.output_path, 'w', encoding='utf8') as d, open('gold.txt', 'w', encoding='utf8') as g:
for i in index:
try:
result = transform(datas[i], schemas[datas[i]['db_id']])
print(result[0])
print('*'*10)
d.write(result[0] + '\n')
g.write("%s\t%s\t%s\n" % (datas[i]['query'], datas[i]["db_id"], datas[i]["question"]))
count += 1