nouhadziri commited on
Commit
898873c
1 Parent(s): ed7f709

README updated

Browse files
Files changed (1) hide show
  1. README.md +39 -0
README.md CHANGED
@@ -1,3 +1,42 @@
1
  ---
2
  license: mit
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ datasets:
4
+ - McGill-NLP/FaithDial
5
+ widget:
6
+ - knowledge: "A cardigan is a type of knitted garment (sweater) that has an open front."
7
+ - response: "The old version is the regular one, knitted garment that has open front and buttons!"
8
  ---
9
+
10
+ ## Overview
11
+
12
+ **Model Description:** roberta-large-faithcritic is the [RoBERTa large model](https://huggingface.co/roberta-large) fine-tuned on 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.
13
+
14
+ The hyperparameters are provided in [hparams.yml](hparams.yml). To know more about how to train a critic model, visit [our repo](https://github.com/McGill-NLP/FaithDial).
15
+
16
+ ## Usage
17
+
18
+ ```python
19
+ from transformers import AutoTokenizer, AutoModel
20
+
21
+ tokenizer = AutoTokenizer.from_pretrained("McGill-NLP/roberta-large-faithcritic")
22
+ model = AutoModel.from_pretrained("McGill-NLP/roberta-large-faithcritic")
23
+
24
+ knowledge = "A cardigan is a type of knitted garment (sweater) that has an open front."
25
+ response = "The old version is the regular one, knitted garment that has open front and buttons!"
26
+ input = tokenizer(knowledge, response)
27
+ output = model(**input)
28
+
29
+ ```
30
+
31
+
32
+ ## Citation Information
33
+
34
+ ```bibtex
35
+ @article{dziri2022faithdial,
36
+ title={FaithDial: A Faithful Benchmark for Information-Seeking Dialogue},
37
+ author={Dziri, Nouha and Kamalloo, Ehsan and Milton, Sivan and Zaiane, Osmar and Yu, Mo and Ponti, Edoardo and Reddy, Siva},
38
+ journal={arXiv preprint, arXiv:2204.10757},
39
+ year={2022},
40
+ url={https://arxiv.org/abs/2204.10757}
41
+ }
42
+ ```