Merge pull request #55 from chamii22:hyp_kg

PiperOrigin-RevId: 307494863

Author: tensorflow-copybara

research/kg_hyp_emb/README.md

@@ -19,16 +19,16 @@ Euclidean embeddings:
 *   CTDecomp [2]
 *   TransE [4]
 *   MurE [5]
-*   RotE (new)
-*   RefE (new)
-*   AttE (new)
+*   RotE [6]
+*   RefE [6]
+*   AttE [6]
 
 Hyperbolic embeddings:
 
-*   TransH (new)
-*   RotH (new)
-*   RefH (new)
-*   AttH (new)
+*   TransH [6]
+*   RotH [6]
+*   RefH [6]
+*   AttH [6]
 
 ## 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 should achieve around 54% Hits@10 on the FB237 test set.
+This model achieves 54% Hits@10 on the FB237 test set.
 
 ## References
 
@@ -93,5 +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.
-Under submission. 2019.
+[6] Chami, Ines, et al. "Low-Dimensional Hyperbolic Knowledge Graph Embeddings."
+Annual Meeting of the Association for Computational Linguistics. 2020.

research/kg_hyp_emb/learning/trainer.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 """Module to train knowledge graph embedding models."""
 
-from kgemb.learning import losses
+from kg_hyp_emb.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 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
+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

research/kg_hyp_emb/models/base.py

@@ -19,7 +19,7 @@
 
 import abc
 
-from kgemb.learning import regularizers
+from kg_hyp_emb.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 kgemb.models.base import KGModel
+from kg_hyp_emb.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 kgemb.models.base import KGModel
-from kgemb.utils import euclidean as euc_utils
+from kg_hyp_emb.models.base import KGModel
+from kg_hyp_emb.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 kgemb.models.base import KGModel
-from kgemb.utils import euclidean as euc_utils
-from kgemb.utils import hyperbolic as hyp_utils
+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
 import numpy as np
 import tensorflow as tf
 

research/kg_hyp_emb/requirements.txt

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

research/kg_hyp_emb/train.py

@@ -29,11 +29,11 @@
 from absl import flags
 from absl import logging
 
-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
+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
 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 kgemb.config import CONFIG
+from kg_hyp_emb.config import CONFIG
 import numpy as np
 
 FLAGS = flags.FLAGS