def build(self,
hiddens=[16],
activations=['relu'],
dropout=0.5,
weight_decay=5e-4,
lr=0.01,
use_bias=False):
if self.backend == "tensorflow":
with tf.device(self.device):
self.model = tfGCN(self.graph.num_node_attrs,
self.graph.num_node_classes,
hiddens=hiddens,
activations=activations,
dropout=dropout,
weight_decay=weight_decay,
lr=lr,
use_bias=use_bias)
else:
self.model = pyGCN(self.graph.num_node_attrs,
self.graph.num_node_classes,
hiddens=hiddens,
activations=activations,
dropout=dropout,
weight_decay=weight_decay,
lr=lr,
use_bias=use_bias).to(self.device)
def build(self,
hiddens=[32],
activations=['relu'],
dropout=0.5,
weight_decay=0.,
lr=0.01,
use_bias=False):
self.model = pyGCN(self.graph.num_node_attrs,
self.graph.num_node_classes,
hiddens=hiddens,
activations=activations,
dropout=dropout,
weight_decay=weight_decay,
lr=lr,
use_bias=use_bias).to(self.device)