# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("drt/wikidata-simplequestions")
model = AutoModelForMaskedLM.from_pretrained("drt/wikidata-simplequestions")Quick Links
A scorer model is trained on Wikidata SimpleQuestions for picking up relevant relations.
The model is trained from roberta-base model.
Check subgraph retrieval toolkit for more details.
- Downloads last month
- 65
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="drt/wikidata-simplequestions")