michal-stefanik
commited on
Commit
•
cbda07a
1
Parent(s):
6ce602e
Update README.md
Browse files
README.md
CHANGED
@@ -92,7 +92,9 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
|
|
92 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
93 |
|
94 |
question = "What is the average, minimum, and maximum age for all French musicians?"
|
95 |
-
schema = ""
|
|
|
|
|
96 |
|
97 |
input_text = " ".join(["Question: ",question, "Schema:", schema])
|
98 |
|
@@ -140,4 +142,6 @@ training_arguments = AdaptationArguments(output_dir="train_dir",
|
|
140 |
num_train_epochs=10,
|
141 |
evaluation_strategy="steps")
|
142 |
|
143 |
-
```
|
|
|
|
|
|
92 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
93 |
|
94 |
question = "What is the average, minimum, and maximum age for all French musicians?"
|
95 |
+
schema = """
|
96 |
+
"stadium" "Stadium_ID" int , "Location" text , "Name" text , "Capacity" int , "Highest" int , "Lowest" int , "Average" int , foreign_key: primary key: "Stadium_ID" [SEP] "singer" "Singer_ID" int , "Name" text , "Country" text , "Song_Name" text , "Song_release_year" text , "Age" int , "Is_male" bool , foreign_key: primary key: "Singer_ID" [SEP] "concert" "concert_ID" int , "concert_Name" text , "Theme" text , "Year" text , foreign_key: "Stadium_ID" text from "stadium" "Stadium_ID" , primary key: "concert_ID" [SEP] "singer_in_concert" foreign_key: "concert_ID" int from "concert" "concert_ID" , "Singer_ID" text from "singer" "Singer_ID" , primary key: "concert_ID" "Singer_ID"
|
97 |
+
"""
|
98 |
|
99 |
input_text = " ".join(["Question: ",question, "Schema:", schema])
|
100 |
|
|
|
142 |
num_train_epochs=10,
|
143 |
evaluation_strategy="steps")
|
144 |
|
145 |
+
```
|
146 |
+
The training is fairly easy to reproduce, but we do not wish to publish modified copies of the Spider datasets that it depends on.
|
147 |
+
If you'd like to investigate further in this direction, feel free to get in touch through a new PR, or via email to nikola.groverova(at)gaussalgo.com.
|