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
- 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