import fed
# Load model
model, tokenizer = fed.load_models("microsoft/DialoGPT-large")
# Evaluate
conversation = "<|endoftext|> Hi! <|endoftext|> Hello, how is your day? <|endoftext|> It's good. It's raining a bit, but I am enjoying a good book. How about you? <|endoftext|> It's good, I just got back from walking my dog What book did you read?"
scores = fed.evaluate(conversation,
model,
tokenizer)
-
Notifications
You must be signed in to change notification settings - Fork 5
Code for SIGdial 2020 paper: Unsupervised Evaluation of Interactive Dialog with DialoGPT (https://arxiv.org/abs/2006.12719)
Shikib/fed
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Code for SIGdial 2020 paper: Unsupervised Evaluation of Interactive Dialog with DialoGPT (https://arxiv.org/abs/2006.12719)
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published