From df3844e733a2e03ccb5f1a72d03a690d7cea1a88 Mon Sep 17 00:00:00 2001 From: EdisonLeeeee Date: Fri, 7 Aug 2020 17:23:27 +0800 Subject: [PATCH] set intx=int32 --- graphgallery/__init__.py | 2 +- graphgallery/config.py | 2 +- graphgallery/nn/models/semisupervised/experimental/s_obvat.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/graphgallery/__init__.py b/graphgallery/__init__.py index 8e0da088..2b344744 100755 --- 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 index 4f925388..c86c3000 100755 --- 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 index 87e9db01..05b14fc0 100755 --- 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):