Skip to content

Commit 003fd71

Browse files
committed
update README
1 parent d411156 commit 003fd71

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,9 @@
33
**Note that this is an early release. Don't hesitate to report bugs/possible improvements! There are surely many.**
44

55

6-
# tibert
6+
# Tibert
77

8-
`tibert` is a transformers-compatible reproduction from the paper [End-to-end Neural Coreference Resolution](https://aclanthology.org/D17-1018/) with several modifications. Among these:
8+
`Tibert` is a transformers-compatible reproduction from the paper [End-to-end Neural Coreference Resolution](https://aclanthology.org/D17-1018/) with several modifications. Among these:
99

1010
- Usage of BERT (or any BERT variant) as an encoder as in [BERT for Coreference Resolution: Baselines and Analysis](https://aclanthology.org/D19-1588/)
1111
- batch size can be greater than 1
@@ -31,11 +31,11 @@ model = BertForCoreferenceResolution.from_pretrained(
3131
)
3232
tokenizer = BertTokenizerFast.from_pretrained("bert-base-cased")
3333

34-
coref_out = predict_coref_simple(
34+
annotated_doc = predict_coref_simple(
3535
"Sli did not want the earpods. He didn't like them.", model, tokenizer
3636
)
3737

38-
pprint_coreference_document(coref_out)
38+
pprint_coreference_document(annotated_doc)
3939
```
4040

4141
results in:

0 commit comments

Comments
 (0)