tdoehmen commited on
Commit
0404bd6
1 Parent(s): 672e437

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -5,7 +5,7 @@ license: cc-by-4.0
5
 
6
  ## Model Description
7
 
8
- In this repository we are introducing **schemapile-fk-starcoder**. It's a language model, based on [google-t5/t5-base](https://huggingface.co/google-t5/t5-base) fine-tuned for predicting foreign key relationships in relational database schemas.
9
 
10
  ## Training Data
11
 
@@ -13,7 +13,7 @@ Forein key pairs extracted from [SchemaPile](https://zenodo.org/records/10931803
13
 
14
  ## Evaluation Data
15
 
16
- We evaluate the foreign key detection accuracy of [schemapile-fk-starcoder](https://huggingface.co/tdoehmen/schemapile-fk-starcoder) and [schemapile-fk-t5](https://huggingface.co/tdoehmen/schemapile-fk-t5) on schemas from [Spider](https://yale-lily.github.io/spider), [BIRD-SQL](https://bird-bench.github.io/), and [CTU PRLR](https://arxiv.org/abs/1511.03086).
17
 
18
  <img src="https://cdn-uploads.huggingface.co/production/uploads/616ea71919594606318887e9/6ouh4u6PFQlY8prLrAm4l.png" alt="eval" width="400"/>
19
 
@@ -48,4 +48,4 @@ Example Output:
48
  'referencedColumn': 'address_id'}
49
  ```
50
 
51
- 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/schemapile-fk-t5-example.ipynb).
 
5
 
6
  ## Model Description
7
 
8
+ In this repository we are introducing **t5-schemapile-fk**. It's a language model, based on [google-t5/t5-base](https://huggingface.co/google-t5/t5-base) fine-tuned for predicting foreign key relationships in relational database schemas.
9
 
10
  ## Training Data
11
 
 
13
 
14
  ## Evaluation Data
15
 
16
+ We evaluate the foreign key detection accuracy of [schemapile-fk-starcoder](https://huggingface.co/tdoehmen/starcoder-schemapile-fk) and [t5-schemapile-fk](https://huggingface.co/tdoehmen/t5-schemapile-fk) on schemas from [Spider](https://yale-lily.github.io/spider), [BIRD-SQL](https://bird-bench.github.io/), and [CTU PRLR](https://arxiv.org/abs/1511.03086).
17
 
18
  <img src="https://cdn-uploads.huggingface.co/production/uploads/616ea71919594606318887e9/6ouh4u6PFQlY8prLrAm4l.png" alt="eval" width="400"/>
19
 
 
48
  'referencedColumn': 'address_id'}
49
  ```
50
 
51
+ 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/t5-schemapile-fk-example.ipynb).