Views
No views yet
from transformers import (
AutoModelForSeq2SeqLM,
AutoTokenizer
)
model_checkpoint = "consciousAI/question-generation-auto-hints-t5-v1-base-s-q-c"
model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
## Input with prompt
context="question_context: <context>"
encodings = tokenizer.encode(context, return_tensors='pt', truncation=True, padding='max_length').to(device)
## You can play with many hyperparams to condition the output, look at demo
output = model.generate(encodings,
#max_length=300,
#min_length=20,
#length_penalty=2.0,
num_beams=4,
#early_stopping=True,
#do_sample=True,
#temperature=1.1
)
## Multiple questions are expected to be delimited by '?' You can write a small wrapper to elegantly format. Look at the demo.
questions = [tokenizer.decode(id, clean_up_tokenization_spaces=False, skip_special_tokens=False) for id in output]| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|---|---|---|---|---|---|---|---|
| 1.9372 | 1.0 | 942 | 1.4811 | 0.5555 | 0.3861 | 0.5243 | 0.5237 |
| 1.2665 | 2.0 | 1884 | 1.4050 | 0.5688 | 0.4056 | 0.5385 | 0.539 |
| 0.955 | 3.0 | 2826 | 1.4131 | 0.5733 | 0.4101 | 0.5426 | 0.5436 |
| 0.7471 | 4.0 | 3768 | 1.4436 | 0.5769 | 0.4179 | 0.5464 | 0.5466 |
| 0.6382 | 5.0 | 4710 | 1.5165 | 0.5819 | 0.4231 | 0.5487 | 0.5491 |