Imported from https://github.com/guotong1988/NL2SQL-BERT