liuyuqi-dellpc
f36d22ed8d
增加 notebook
|
4 years ago | |
|---|---|---|
| .vscode | 4 years ago | |
| data | 4 years ago | |
| img | 4 years ago | |
| saved_model | 4 years ago | |
| sqlnet | 4 years ago | |
| .dockerignore | 4 years ago | |
| .gitignore | 4 years ago | |
| Dockerfile | 4 years ago | |
| LICENSE | 4 years ago | |
| README.cn.md | 4 years ago | |
| README.md | 4 years ago | |
| colab-nl2sql.ipynb | 4 years ago | |
| docker-compose.debug.yml | 4 years ago | |
| docker-compose.yml | 4 years ago | |
| pip.conf | 4 years ago | |
| requirements.txt | 4 years ago | |
| start_test.sh | 4 years ago | |
| start_train.sh | 4 years ago | |
| test.py | 4 years ago | |
| train.py | 4 years ago |
README.cn.md
nl2sql 文字问题转 sql
https://drive.google.com/open?id=10RtAom_D4zOp_w5OsYLTtLd2TEuC7C1X
模型训练所需要的算力还是挺大的，在 K80 上训练一个 Epoch 大概需要 7 到 8 分钟，基线模型默认训练 100 个 Epoch。追一表示试验多次后，验证集上的平均 Logic Form Accuracy 为 27.5%，即 SQL 每一个子句全都预测正确的概率为 27.5%。
WikiSQL 基准：https://github.com/salesforce/WikiSQL
CUDA_VISIBLE_DEVICES=0 python train.py --ca --gpu --bs 128