763337092 / MLMLM

MLMLM: Link Prediction with Mean Likelihood Masked Language Model

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

How to run

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

cd ./code
CUDA_VISIBLE_DEVICES=0 python main.py