fixed import errors

Author: ines-chami

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/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