Skip to content

Commit 34cee3a

Browse files
committed
added citation to README
1 parent 9711cbc commit 34cee3a

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

research/kg_hyp_emb/README.md

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -19,16 +19,16 @@ Euclidean embeddings:
1919
* CTDecomp [2]
2020
* TransE [4]
2121
* MurE [5]
22-
* RotE (new)
23-
* RefE (new)
24-
* AttE (new)
22+
* RotE [6]
23+
* RefE [6]
24+
* AttE [6]
2525

2626
Hyperbolic embeddings:
2727

28-
* TransH (new)
29-
* RotH (new)
30-
* RefH (new)
31-
* AttH (new)
28+
* TransH [6]
29+
* RotH [6]
30+
* RefH [6]
31+
* AttH [6]
3232

3333
## Usage
3434

@@ -73,7 +73,7 @@ RefE on FB15k-237:
7373
python train.py --max_epochs 100 --dataset FB237 --model RefE --loss_fn SigmoidCrossEntropy --neg_sample_size -1 --data_dir data --optimizer Adagrad --lr 5e-2 --save_dir logs --rank 500 --entity_reg 1e-5 --rel_reg 1e-5 --patience 10 --valid 5 --save_model=false --save_logs=true --regularizer L3 --initializer GlorotNormal
7474
```
7575

76-
This model should achieve around 54% Hits@10 on the FB237 test set.
76+
This model achieves 54% Hits@10 on the FB237 test set.
7777

7878
## References
7979

@@ -93,5 +93,4 @@ multi-relational data." Advances in neural information processing systems. 2013.
9393
[5] Balažević, Ivana, et al. "Multi-relational Poincaré Graph Embeddings."
9494
Advances in neural information processing systems. 2019.
9595

96-
[6] Chami, Ines, et al. Low-Dimensional Hyperbolic Knowledge Graph Embeddings.
97-
Under submission. 2019.
96+
[6] Chami, Ines, et al. "Low-Dimensional Hyperbolic Knowledge Graph Embeddings." Annual Meeting of the Association for Computational Linguistics. 2020.

0 commit comments

Comments
 (0)