Update README.md
Browse files
README.md
CHANGED
@@ -103,4 +103,21 @@ print(responses)
|
|
103 |
|
104 |
You can also fine-tune Prem-1B-SQL using HuggingFace Transformers and with [PremSQL Tuners](https://docs.premai.io/premsql/tuners) as well.
|
105 |
Please [check out our documentation](https://docs.premai.io/premsql) to know about more about PremSQL and all the features
|
106 |
-
we provide.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
103 |
|
104 |
You can also fine-tune Prem-1B-SQL using HuggingFace Transformers and with [PremSQL Tuners](https://docs.premai.io/premsql/tuners) as well.
|
105 |
Please [check out our documentation](https://docs.premai.io/premsql) to know about more about PremSQL and all the features
|
106 |
+
we provide.
|
107 |
+
|
108 |
+
|
109 |
+
## Datasets used to train the model
|
110 |
+
|
111 |
+
Prem-1B-SQL is trained using the following datasets:
|
112 |
+
|
113 |
+
1. [BirdBench Training dataset](https://bird-bench.github.io/) | Uploaded on [PremSQL datasets on HF](https://huggingface.co/datasets/premai-io/birdbench)
|
114 |
+
2. [Spider dataset](https://yale-lily.github.io/spider) | Uploaded on [PremSQL datasets on HF](https://huggingface.co/datasets/premai-io/spider)
|
115 |
+
3. [Domain specialization dataset, gathered and uploaded to PremSQL datasets](https://huggingface.co/datasets/premai-io/domains)
|
116 |
+
4. [Gretel AI synthetic dataset](https://huggingface.co/datasets/gretelai/synthetic_text_to_sql?row=0)
|
117 |
+
|
118 |
+
Additionally we made error handling datasets on top of these datasets to make the model learn from its errors and self correct them.
|
119 |
+
|
120 |
+
|
121 |
+
## Evaluation results of Prem-1B-SQL
|
122 |
+
|
123 |
+
The results of Prem-1B-SQL on some public benchmarks will be published soon.
|