Update README.md
Browse files
README.md
CHANGED
|
@@ -113,4 +113,19 @@ with torch.no_grad():
|
|
| 113 |
if generated_answer == "Yes":
|
| 114 |
print("[Score]: ", yes_prob)
|
| 115 |
elif generated_answer == "No":
|
| 116 |
-
print("[Score]: ", 0)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 113 |
if generated_answer == "Yes":
|
| 114 |
print("[Score]: ", yes_prob)
|
| 115 |
elif generated_answer == "No":
|
| 116 |
+
print("[Score]: ", 0)
|
| 117 |
+
```
|
| 118 |
+
|
| 119 |
+
If you use **GradeSQL** in your research, please cite the following paper:
|
| 120 |
+
|
| 121 |
+
```bibtex
|
| 122 |
+
@misc{gradesqloutcomerewardmodels2025,
|
| 123 |
+
title={GradeSQL: Outcome Reward Models for Ranking SQL Queries from Large Language Models},
|
| 124 |
+
author={Mattia Tritto and Giuseppe Farano and Dario Di Palma and Gaetano Rossiello and Fedelucio Narducci and Dharmashankar Subramanian and Tommaso Di Noia},
|
| 125 |
+
year={2025},
|
| 126 |
+
eprint={2509.01308},
|
| 127 |
+
archivePrefix={arXiv},
|
| 128 |
+
primaryClass={cs.AI},
|
| 129 |
+
url={https://arxiv.org/abs/2509.01308},
|
| 130 |
+
}
|
| 131 |
+
```
|