Aunsiels/InfantBooks
Viewer • Updated • 496 • 106 • 7
How to use Aunsiels/ChildBERT with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="Aunsiels/ChildBERT") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("Aunsiels/ChildBERT")
model = AutoModelForMaskedLM.from_pretrained("Aunsiels/ChildBERT")A BERT-model finetuned on children's books.
Romero, J., & Razniewski, S. (2022).
Do Children Texts Hold The Key To Commonsense Knowledge?
In Proceedings of the 2022 Joint Conference on Empirical Methods in Natural Language Processing and Computational Natural Language Learning.