SelfCheckGPT: Assessing text-based responses from LLMs
Project description
Code for our paper “SelfCheckGPT: Zero-Resource Black-Box Hallucination Detection for Generative Large Language Models”, https://arxiv.org/abs/2303.08896
More information can be found on our project page: https://github.com/potsawee/selfcheckgpt
Installation
pip install selfcheckgpt
SelfCheckGPT Usage
See more details in Jupyter Notebook: https://github.com/potsawee/selfcheckgpt/blob/main/demo/SelfCheck_demo1.ipynb
from selfcheckgpt.modeling_selfcheck import SelfCheckMQAG, SelfCheckBERTScore
selfcheck_mqag = SelfCheckMQAG()
selfcheck_bertscore = SelfCheckBERTScore()
sent_scores_mqag = selfcheck_mqag.predict(
sentences,
passage,
[sample1, sample2, sample3],
num_questions_per_sent = 5,
scoring_method = 'bayes_with_alpha',
beta1 = 0.8, beta2 = 0.8,
)
sent_scores_bertscore = selfcheck_bertscore.predict(
sentences,
[sample1, sample2, sample3],
)
MQAG Usage
See more details in Jupyter Notebook: https://github.com/potsawee/selfcheckgpt/blob/main/demo/MQAG_demo1.ipynb
from selfcheckgpt.modeling_mqag import MQAG
mqag_model = MQAG()
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
selfcheckgpt-0.1.2.tar.gz
(10.3 kB
view hashes)