nouhadziri's picture
Update README.md
ef69439
---
license: mit
datasets:
- McGill-NLP/FaithDial
widget:
- text: "A cardigan is a type of knitted garment (sweater) that has an open front. </s></s> The old version is the regular one, knitted garment that has open front and buttons!"
model-index:
- name: roberta-large-faithcritic
results:
- task:
type: text-classification
name: Faithfulness Critic
dataset:
name: 'FaithCritic'
type: McGill-NLP/FaithDial
metrics:
- name: Accuracy
type: accuracy
value: 86.51
- task:
type: text-classification
name: Faithfulness Critic
dataset:
name: 'MNLI'
type: multi_nli
metrics:
- name: Accuracy
type: accuracy
value: 74.720
- task:
type: text-classification
name: Faithfulness Critic
dataset:
name: 'BEGIN'
type: begin
metrics:
- name: Accuracy
type: accuracy
value: 71.56
---
## Overview
**Model Description:** roberta-large-faithcritic is the [RoBERTa large model](https://huggingface.co/roberta-large) fine-tuned on FaithCritic, a derivative of the [FaithDial](https://huggingface.co/datasets/McGill-NLP/FaithDial) dataset. The objective is to predict whether an utterance is faithful or not, given the source knowledge.
The hyperparameters are provided in [hparams.yml](https://huggingface.co/McGill-NLP/roberta-large-faithcritic/blob/main/hparams.yaml). To know more about how to train a critic model, visit [our repo](https://github.com/McGill-NLP/FaithDial).
## Usage
```python
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("McGill-NLP/roberta-large-faithcritic")
model = AutoModel.from_pretrained("McGill-NLP/roberta-large-faithcritic")
knowledge = "A cardigan is a type of knitted garment (sweater) that has an open front."
response = "The old version is the regular one, knitted garment that has open front and buttons!"
input = tokenizer(knowledge, response)
output = model(**input)
```
## Citation Information
```bibtex
@article{dziri2022faithdial,
title={FaithDial: A Faithful Benchmark for Information-Seeking Dialogue},
author={Dziri, Nouha and Kamalloo, Ehsan and Milton, Sivan and Zaiane, Osmar and Yu, Mo and Ponti, Edoardo and Reddy, Siva},
journal={arXiv preprint, arXiv:2204.10757},
year={2022},
url={https://arxiv.org/abs/2204.10757}
}
```