update readme

peterwan1 · Sep 4, 2019 · b03203c · b03203c
1 parent 62b76ed
commit b03203c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -1,6 +1,7 @@
 # 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
 
@@ -80,7 +81,7 @@ python3 run_bert_relation_prediction.py
 --eval_batch_size 512
 ```
 
-### 2. Link Prediction
+### 3. Link Prediction
 
 #### WN18RR
 

diff --git a/run_bert_triple_classifier.py b/run_bert_triple_classifier.py
@@ -117,7 +117,7 @@ def __init__(self):
     def get_train_examples(self, data_dir):
         """See base class."""
         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):
         """See base class."""