MLMLM

MLMLM: Link Prediction with Mean Likelihood Masked Language Model

Paper	Official code	Note
MLMLM: Link Prediction with Mean Likelihood Masked Language Model	Code	Note

Environment

python3.74
pytorch>=1.3.1
transformers==2.8.0
numpy
pandas
tqdm

download bert-base-uncased weights from https://huggingface.co/bert-base-uncased/tree/main to ./pretrained/bert-base-uncased/
run following commands

cd ./code
CUDA_VISIBLE_DEVICES=0 python main.py

Name		Name	Last commit message	Last commit date
Latest commit History 17 Commits
code		code
data		data
pretrained/bert-base-uncased		pretrained/bert-base-uncased
README.md		README.md