anindya64 commited on
Commit
fd83f4b
·
verified ·
1 Parent(s): ba90f9c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -1
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.