KG/kg-bert
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update readme

Browse Source
This commit is contained in:
Liang Yao 2019-09-04 07:43:21 -05:00
parent 62b76ed565
commit b03203c1a0
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@ -1,6 +1,7 @@
# KG-BERT: BERT for Knowledge Graph Completion # KG-BERT: BERT for Knowledge Graph Completion
The repository is modified from [pytorch-pretrained-BERT](https://github.com/huggingface/pytorch-pretrained-BERT). The repository is modified from [pytorch-pretrained-BERT](https://github.com/huggingface/pytorch-pretrained-BERT) and tested on Python 3.5+.
## Installing requirement packages ## Installing requirement packages
@ -80,7 +81,7 @@ python3 run_bert_relation_prediction.py
--eval_batch_size 512 --eval_batch_size 512
``` ```
### 2. Link Prediction ### 3. Link Prediction
#### WN18RR #### WN18RR

2
run_bert_triple_classifier.py
View File

@ -117,7 +117,7 @@ class KGProcessor(DataProcessor):
def get_train_examples(self, data_dir): def get_train_examples(self, data_dir):
"""See base class.""" """See base class."""
return self._create_examples( return self._create_examples(
self._read_tsv(os.path.join(data_dir, "train_10.tsv")), "train", data_dir) self._read_tsv(os.path.join(data_dir, "train.tsv")), "train", data_dir)
def get_dev_examples(self, data_dir): def get_dev_examples(self, data_dir):
"""See base class.""" """See base class."""