Views
No views yet

wikisql from Huggingface/NLP| Dataset | Split | # samples |
|---|---|---|
| wikisql | train | 56355 |
| wikisql | valid | 14436 |
1train_dataset = nlp.load_dataset('wikisql', split=nlp.Split.TRAIN)
2valid_dataset = nlp.load_dataset('wikisql', split=nlp.Split.VALIDATION)1from transformers import AutoModelWithLMHead, AutoTokenizer
2
3tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL-sql-to-en")
4model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL-sql-to-en")
5
6def get_explanation(query):
7 input_text = "translate Sql to English: %s </s>" % query
8 features = tokenizer([input_text], return_tensors='pt')
9
10 output = model.generate(input_ids=features['input_ids'],
11 attention_mask=features['attention_mask'])
12
13 return tokenizer.decode(output[0])
14
15query = "SELECT COUNT Params form model where location=HF-Hub"
16
17get_explanation(query)
18
19# output: 'How many parameters form model for HF-hub?'Created by Manuel Romero/@mrm8488 | LinkedIn
Made with ♥ in Spain