rakeshkiriyath
commited on
Commit
·
68c125e
1
Parent(s):
748ac9b
Edited ReadME
Browse files
README.md
CHANGED
@@ -9,7 +9,7 @@ tags:
|
|
9 |
- text2sql
|
10 |
- sql
|
11 |
---
|
12 |
-
# Model Card
|
13 |
|
14 |
<!-- The base model used for training is gpt2-medium. We finetuned it on the following dataset: b-mc2/sql-create-context -->
|
15 |
|
@@ -26,8 +26,6 @@ f"Translate the following English question to SQL: {query}
|
|
26 |
|
27 |
SELECT zip_code FROM Visibility WHERE AVG(visibility) < 10
|
28 |
|
29 |
-
[More Information Needed]
|
30 |
-
|
31 |
|
32 |
#### Training Hyperparameters
|
33 |
|
@@ -40,12 +38,12 @@ weight_decay=0.01
|
|
40 |
|
41 |
## Evaluation
|
42 |
|
43 |
-
Step
|
44 |
-
|
45 |
-
500
|
46 |
-
1000
|
47 |
-
1500
|
48 |
-
2000
|
49 |
|
50 |
{'eval_loss': 0.23689331114292145, 'eval_runtime': 104.4102, 'eval_samples_per_second': 67.043, 'eval_steps_per_second': 8.38, 'epoch': 1.0}
|
51 |
|
|
|
9 |
- text2sql
|
10 |
- sql
|
11 |
---
|
12 |
+
# Model Card
|
13 |
|
14 |
<!-- The base model used for training is gpt2-medium. We finetuned it on the following dataset: b-mc2/sql-create-context -->
|
15 |
|
|
|
26 |
|
27 |
SELECT zip_code FROM Visibility WHERE AVG(visibility) < 10
|
28 |
|
|
|
|
|
29 |
|
30 |
#### Training Hyperparameters
|
31 |
|
|
|
38 |
|
39 |
## Evaluation
|
40 |
|
41 |
+
| Step | Training Loss |
|
42 |
+
| -------- | ------- |
|
43 |
+
| 500 | 0.337800 |
|
44 |
+
| 1000 | 0.262900 |
|
45 |
+
| 1500 | 0.253200 |
|
46 |
+
| 2000 | 0.246400 |
|
47 |
|
48 |
{'eval_loss': 0.23689331114292145, 'eval_runtime': 104.4102, 'eval_samples_per_second': 67.043, 'eval_steps_per_second': 8.38, 'epoch': 1.0}
|
49 |
|