set intx=int32

EdisonLeeeee · Aug 7, 2020 · df3844e · df3844e
1 parent 3ba050e
commit df3844e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/graphgallery/__init__.py b/graphgallery/__init__.py
@@ -6,7 +6,7 @@
                                            astensors)
 
 from graphgallery.utils.data_utils import normalize_adj, normalize_x, Bunch, sample_mask
-from graphgallery.config import set_epsilon, set_floatx, set_intx, floatx, intx
+from graphgallery.config import set_epsilon, set_floatx, set_intx, epsilon, floatx, intx
 from graphgallery.utils.tensor_utils import normalize_adj_tensor, normalize_edge_tensor
 from graphgallery.utils.shape_utils import repeat
 from graphgallery.utils.type_check import *

diff --git a/graphgallery/config.py b/graphgallery/config.py
@@ -10,7 +10,7 @@
 epsilon = K.epsilon
 set_epsilon = K.set_epsilon
 
-_INTX = 'int64'
+_INTX = 'int32'
 
 def intx():
     """Returns the default integer type, as a string.

diff --git a/graphgallery/nn/models/semisupervised/experimental/s_obvat.py b/graphgallery/nn/models/semisupervised/experimental/s_obvat.py
@@ -77,7 +77,7 @@ def preprocess(self, adj, x):
         with tf.device(self.device):
             self.x_norm, self.adj_norm = astensors([x, adj])
 
-    def build(self, hiddens=[16], activations=['relu'], dropouts=[0.5],
+    def build(self, hiddens=[16], activations=['relu'], dropouts=[0.],
               lr=0.01, l2_norms=[5e-4], p1=1.4, p2=0.7, use_bias=False,
               epsilon=0.01):