Views
No views yet
1from classifier import Classifier, get_relevance_discriminator, get_nli_discriminator
2from score import score_apc, score_APC
3
4relevance_discriminator = get_relevance_discriminator(character=None, statement_query_relevance_dataset=None, relevance_finetune_epoch=None, use_pretrained_discriminator=True)
5nli_discriminator = get_nli_discriminator(character=None, statement_to_response_nli_v2_dataset=None, nli_finetune_epoch=None, use_pretrained_discriminator=True)
6
7character = "Komeiji Koishi"
8statements = ["Komeiji Koishi lives with her sister, Komeiji Satori.", "Komeiji Koishi lives in Chireiden."]
9query = "Where do you live, Koishi?"
10responses = ["I live in Chireiden with my sister, Satori!", "I live in Chireiden!", "I live in Hakurei Shrine!"]
11print([score_APC(character, statements, query, response, relevance_discriminator, nli_discriminator).item() for response in responses])
12
13# [1.6079180240631104, 0.9955980777740479, 0.03315635025501251]