Method
__init__
(self, conv, flow, dims,
fanouts, metapath,
add_self_loops=True,
max_id=-1,
**kwargs)
Source from the content-addressed store, hash-verified
| 27 | class BaseGNNNet(object): |
| 28 | |
| 29 | def __init__(self, conv, flow, dims, |
| 30 | fanouts, metapath, |
| 31 | add_self_loops=True, |
| 32 | max_id=-1, |
| 33 | **kwargs): |
| 34 | conv_class = utils.get_conv_class(conv) |
| 35 | flow_class = utils.get_flow_class(flow) |
| 36 | if flow_class == 'whole': |
| 37 | self.whole_graph = True |
| 38 | else: |
| 39 | self.whole_graph = False |
| 40 | self.convs = [] |
| 41 | for dim in dims[:-1]: |
| 42 | self.convs.append(self.get_conv(conv_class, dim)) |
| 43 | self.fc = tf.layers.Dense(dims[-1]) |
| 44 | self.sampler = flow_class(fanouts, metapath, add_self_loops, max_id=max_id) |
| 45 | |
| 46 | def get_conv(self, conv_class, dim): |
| 47 | return conv_class(dim) |
Callers
nothing calls this directly
Tested by
no test coverage detected