Add pipeline tag, library name, link to project page, and link to paper
#1
by
nielsr
HF Staff
- opened
README.md
CHANGED
|
@@ -1,8 +1,15 @@
|
|
| 1 |
---
|
| 2 |
license: cc-by-4.0
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
| 4 |
# SchemaPile Foreign Key Detection Model (Starcoder)
|
| 5 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
## Model Description
|
| 7 |
|
| 8 |
In this repository we are introducing **starcoder-schemapile-fk**. It's a language model, based on [BigCode/starcoder](https://huggingface.co/bigcode/starcoder) fine-tuned for predicting foreign key relationships in relational database schemas.
|
|
@@ -51,4 +58,4 @@ Example Output:
|
|
| 51 |
'referencedColumn': 'address_id'}
|
| 52 |
```
|
| 53 |
|
| 54 |
-
To run the model locally, we recommend using our end-to-end [Example Notebook](https://github.com/amsterdata/schemapile/blob/main/experiments/foreign_key_detection/starcoder-schemapile-fk-example.ipynb) (requires a single A100 40GB).
|
|
|
|
| 1 |
---
|
| 2 |
license: cc-by-4.0
|
| 3 |
+
pipeline_tag: table-question-answering
|
| 4 |
+
library_name: transformers
|
| 5 |
---
|
| 6 |
+
|
| 7 |
# SchemaPile Foreign Key Detection Model (Starcoder)
|
| 8 |
|
| 9 |
+
The goal of this repository is to support machine learning research with multi-relational data, as outlined in the paper [The CTU Prague Relational Learning Repository](https://huggingface.co/papers/1511.03086).
|
| 10 |
+
|
| 11 |
+
Project Page: https://relational-data.org
|
| 12 |
+
|
| 13 |
## Model Description
|
| 14 |
|
| 15 |
In this repository we are introducing **starcoder-schemapile-fk**. It's a language model, based on [BigCode/starcoder](https://huggingface.co/bigcode/starcoder) fine-tuned for predicting foreign key relationships in relational database schemas.
|
|
|
|
| 58 |
'referencedColumn': 'address_id'}
|
| 59 |
```
|
| 60 |
|
| 61 |
+
To run the model locally, we recommend using our end-to-end [Example Notebook](https://github.com/amsterdata/schemapile/blob/main/experiments/foreign_key_detection/starcoder-schemapile-fk-example.ipynb) (requires a single A100 40GB).
|