Skip to content

Commit d6ca78b

Browse files
authored
Update the table in README.md
1 parent 617062a commit d6ca78b

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

README.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -64,15 +64,16 @@ The run.sh script provides an easy way to search hyper-parameters:
6464
**Speed**
6565

6666
The KGE models usually take about half an hour to run 10000 steps on a single GeForce GTX 1080 Ti GPU with default configuration. And these models need different max_steps to converge on different data sets:
67-
|Dataset| FB15k | FB15k-237 | wn18 | wn18rr | Countries S* |
68-
|--|--|--|--|--|--|
67+
68+
| Dataset | FB15k | FB15k-237 | wn18 | wn18rr | Countries S* |
69+
|-------------|-------------|-------------|-------------|-------------|-------------|
6970
|MAX_STEPS| 150000 | 100000 | 80000 | 80000 | 40000 |
7071
|TIME| 9 h | 6 h | 4 h | 4 h | 2 h |
7172

7273
**Results of the RotatE model**
7374

74-
|| FB15k | FB15k-237 | wn18 | wn18rr |
75-
|--|--|--|--|--|--|
75+
| Dataset | FB15k | FB15k-237 | wn18 | wn18rr |
76+
|-------------|-------------|-------------|-------------|-------------|
7677
| MRR | .797 ± .001 | .949 ± .000 |.337 ± .001 |.477 ± .001
7778
| MR | 40 | 177 | 309 | 3340 |
7879
| HITS@1 | .746 | .241 | .944 | .428 |

0 commit comments

Comments
 (0)