fix wrong input used

nalexai · Feb 26, 2024 · 586830d · 586830d
1 parent 275eeb7
commit 586830d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/hyperlib/models/__init__.py b/hyperlib/models/__init__.py
@@ -40,7 +40,7 @@ def call(self, inputs):
         return self.embedding(indices)
 
     def get_embeddings(self):
-        embeddings = self.embedding (tf.constant([i for i in range(len(self.vocab))]))
+        embeddings = self.embedding(tf.constant([i for i in range(len(self.vocab))]))
         embeddings_copy = tf.identity(embeddings)
         embeddings_hyperbolic = self.manifold.expmap0(embeddings_copy, c=self.c)
         return embeddings_hyperbolic

diff --git a/hyperlib/nn/layers/graph.py b/hyperlib/nn/layers/graph.py
@@ -27,8 +27,8 @@ def __init__(self, manifold, input_size, c, activation):
 
         self.manifold = manifold
         self.c = tf.Variable([c], trainable=False)
-        #self.linear_layer = LinearHyperbolic(input_size, self.manifold, self.c, activation=None)
-        self.linear_layer = LinearHyperbolic(1433, self.manifold, 1.0, activation=None)
+        self.linear_layer = LinearHyperbolic(input_size, self.manifold, self.c, activation=None)
+        #self.linear_layer = LinearHyperbolic(1433, self.manifold, 1.0, activation=None)
         self.aggregation_layer = HyperbolicAggregation(self.manifold, self.c)
         self.activation_layer = ActivationHyperbolic(self.manifold, self.c, self.c, activation)