(self, conv, flow,
dims, fanouts, metapath,
feature_idx, feature_dim,
add_self_loops=False)
| 27 | class GNN(BaseGNNNet): |
| 28 | |
| 29 | def __init__(self, conv, flow, |
| 30 | dims, fanouts, metapath, |
| 31 | feature_idx, feature_dim, |
| 32 | add_self_loops=False): |
| 33 | super(GNN, self).__init__(conv=conv, |
| 34 | flow=flow, |
| 35 | dims=dims, |
| 36 | fanouts=fanouts, |
| 37 | metapath=metapath, |
| 38 | add_self_loops=add_self_loops) |
| 39 | if not isinstance(feature_idx, list): |
| 40 | feature_idx = [feature_idx] |
| 41 | if not isinstance(feature_dim, list): |
| 42 | feature_dim = [feature_dim] |
| 43 | self.feature_idx = feature_idx |
| 44 | self.feature_dim = feature_dim |
| 45 | |
| 46 | def to_x(self, n_id): |
| 47 | x, = tf_euler.get_dense_feature(n_id, |
nothing calls this directly
no outgoing calls
no test coverage detected