internal

PiperOrigin-RevId: 307136389

Author: Neural-Link Team

research/kg_hyp_emb/README.md

@@ -19,16 +19,16 @@ Euclidean embeddings:
 *   CTDecomp [2]
 *   TransE [4]
 *   MurE [5]
-*   RotE [6]
-*   RefE [6]
-*   AttE [6]
+*   RotE (new)
+*   RefE (new)
+*   AttE (new)
 
 Hyperbolic embeddings:
 
-*   TransH [6]
-*   RotH [6]
-*   RefH [6]
-*   AttH [6]
+*   TransH (new)
+*   RotH (new)
+*   RefH (new)
+*   AttH (new)
 
 ## Usage
 
@@ -73,7 +73,7 @@ RefE on FB15k-237:
 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
 ```
 
-This model achieves 54% Hits@10 on the FB237 test set.
+This model should achieve around 54% Hits@10 on the FB237 test set.
 
 ## References
 
@@ -93,4 +93,5 @@ multi-relational data." Advances in neural information processing systems. 2013.
 [5] Balažević, Ivana, et al. "Multi-relational Poincaré Graph Embeddings."
 Advances in neural information processing systems. 2019.
 
-[6] Chami, Ines, et al. "Low-Dimensional Hyperbolic Knowledge Graph Embeddings." Annual  Meeting  of  the  Association  for Computational Linguistics. 2020.
+[6] Chami, Ines, et al. Low-Dimensional Hyperbolic Knowledge Graph Embeddings.
+Under submission. 2019.

research/kg_hyp_emb/learning/trainer.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 """Module to train knowledge graph embedding models."""
 
-from kg_hyp_emb.learning import losses
+from kgemb.learning import losses
 import tensorflow as tf
 
 

research/kg_hyp_emb/models/__init__.py

@@ -13,15 +13,15 @@
 # limitations under the License.
 """Knowledge graph embedding models."""
 
-from kg_hyp_emb.models.complex import Complex
-from kg_hyp_emb.models.complex import RotatE
-from kg_hyp_emb.models.euclidean import AttE
-from kg_hyp_emb.models.euclidean import CTDecomp
-from kg_hyp_emb.models.euclidean import MurE
-from kg_hyp_emb.models.euclidean import RefE
-from kg_hyp_emb.models.euclidean import RotE
-from kg_hyp_emb.models.euclidean import TransE
-from kg_hyp_emb.models.hyperbolic import AttH
-from kg_hyp_emb.models.hyperbolic import RefH
-from kg_hyp_emb.models.hyperbolic import RotH
-from kg_hyp_emb.models.hyperbolic import TransH
+from kgemb.models.complex import Complex
+from kgemb.models.complex import RotatE
+from kgemb.models.euclidean import AttE
+from kgemb.models.euclidean import CTDecomp
+from kgemb.models.euclidean import MurE
+from kgemb.models.euclidean import RefE
+from kgemb.models.euclidean import RotE
+from kgemb.models.euclidean import TransE
+from kgemb.models.hyperbolic import AttH
+from kgemb.models.hyperbolic import RefH
+from kgemb.models.hyperbolic import RotH
+from kgemb.models.hyperbolic import TransH

research/kg_hyp_emb/models/base.py

@@ -19,7 +19,7 @@
 
 import abc
 
-from kg_hyp_emb.learning import regularizers
+from kgemb.learning import regularizers
 import numpy as np
 import tensorflow as tf
 

research/kg_hyp_emb/models/complex.py

@@ -17,7 +17,7 @@
 from __future__ import division
 from __future__ import print_function
 
-from kg_hyp_emb.models.base import KGModel
+from kgemb.models.base import KGModel
 import tensorflow as tf
 
 

research/kg_hyp_emb/models/euclidean.py

@@ -17,8 +17,8 @@
 from __future__ import division
 from __future__ import print_function
 
-from kg_hyp_emb.models.base import KGModel
-from kg_hyp_emb.utils import euclidean as euc_utils
+from kgemb.models.base import KGModel
+from kgemb.utils import euclidean as euc_utils
 import numpy as np
 import tensorflow as tf
 

research/kg_hyp_emb/models/hyperbolic.py

@@ -17,9 +17,9 @@
 from __future__ import division
 from __future__ import print_function
 
-from kg_hyp_emb.models.base import KGModel
-from kg_hyp_emb.utils import euclidean as euc_utils
-from kg_hyp_emb.utils import hyperbolic as hyp_utils
+from kgemb.models.base import KGModel
+from kgemb.utils import euclidean as euc_utils
+from kgemb.utils import hyperbolic as hyp_utils
 import numpy as np
 import tensorflow as tf
 

research/kg_hyp_emb/requirements.txt

@@ -1,3 +1,2 @@
-absl-py
-numpy
-tensorflow-gpu==2.0
+numpy>=1.17
+tensorflow_gpu>=2.0.0

research/kg_hyp_emb/train.py

@@ -29,11 +29,11 @@
 from absl import flags
 from absl import logging
 
-from kg_hyp_emb.config import CONFIG
-from kg_hyp_emb.datasets.datasets import DatasetFn
-from kg_hyp_emb.learning.trainer import KGTrainer
-import kg_hyp_emb.models as models
-import kg_hyp_emb.utils.train as train_utils
+from kgemb.config import CONFIG
+from kgemb.datasets.datasets import DatasetFn
+from kgemb.learning.trainer import KGTrainer
+import kgemb.models as models
+import kgemb.utils.train as train_utils
 import tensorflow as tf
 
 flag_fns = {

research/kg_hyp_emb/utils/train.py

@@ -18,7 +18,7 @@
 from __future__ import print_function
 
 from absl import flags
-from kg_hyp_emb.config import CONFIG
+from kgemb.config import CONFIG
 import numpy as np
 
 FLAGS = flags.FLAGS