/nl2sql

阿里天池首届中文NL2SQL挑战赛top6

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

排名: 6

队名: 爆写规则一万行

成员: eguilg, 严之zh, naniwet

Environments

Ubuntu 18.04

Python: 3.6.5

Pytorch: 1.1.0

CUDA: 9.0

CUDNN: 7.1.3

Required packages

We used pytorch-pretrained-bert package for backbone BERT model.

(Note that the original pytorch-pretrained-bert was updated to pytorch-transformers during the contest, but we chose to stick to the old version for stability.)

Required Python packages:

fuzzywuzzy==0.17.0
numpy==1.17.0
torch==1.1.0
pytorch-pretrained-bert==0.6.2
tqdm==4.24.0
records 

Command to install the required python packages:

pip install -r requirements.txt