search

yao8839836 / kg-bert

KG-BERT: BERT for Knowledge Graph Completion
Apache License 2.0
679 stars 141 forks source link

commands not working for prediction #18

Closed ChloeJKim closed 3 years ago

ChloeJKim commented 3 years ago

Hi Thanks for posting your code! love your work and wanted to reproduce the result.

For 3. Link Prediction (UMLS) I ran the exact same command as you posted but get this. Can you help me how I can fix this?

python3 run_bert_link_prediction.py --task_name kg
--do_train
--do_eval --do_predict --data_dir ./data/umls --bert_model bert-base-uncased --max_seq_length 15 --train_batch_size 32 --learning_rate 5e-5 --num_train_epochs 5.0 --output_dir ./output_umls/
--gradient_accumulation_steps 1 --eval_batch_size 135

image

Thank you in advance

yao8839836 commented 3 years ago

@ChloeJKim

Thank you! The command should be in the same command line.