(self, conv, flow,
dims, fanouts, metapath,
feature_idx, feature_dim,
add_self_loops=False)
| 52 | class GNN(BaseGNNNet): |
| 53 | |
| 54 | def __init__(self, conv, flow, |
| 55 | dims, fanouts, metapath, |
| 56 | feature_idx, feature_dim, |
| 57 | add_self_loops=False): |
| 58 | super(GNN, self).__init__(conv=conv, |
| 59 | flow=flow, |
| 60 | dims=dims, |
| 61 | fanouts=fanouts, |
| 62 | metapath=metapath, |
| 63 | add_self_loops=add_self_loops) |
| 64 | if not isinstance(feature_idx, list): |
| 65 | feature_idx = [feature_idx] |
| 66 | if not isinstance(feature_dim, list): |
| 67 | feature_dim = [feature_dim] |
| 68 | self.feature_idx = feature_idx |
| 69 | self.feature_dim = feature_dim |
| 70 | |
| 71 | def to_x(self, n_id): |
| 72 | x, = tf_euler.get_dense_feature(n_id, |
nothing calls this directly
no outgoing calls
no test coverage detected