(dim, num_layers, aggregator, **kwargs)
| 416 | |
| 417 | @staticmethod |
| 418 | def create_aggregators(dim, num_layers, aggregator, **kwargs): |
| 419 | new_aggregators = [] |
| 420 | aggregator_class = aggregators.get(aggregator) |
| 421 | for layer in range(num_layers): |
| 422 | activation = tf.nn.relu if layer < num_layers - 1 else None |
| 423 | new_aggregators.append( |
| 424 | aggregator_class(dim, activation=activation, **kwargs)) |
| 425 | return new_aggregators |
| 426 | |
| 427 | def __init__(self, metapath, fanouts, dim, |
| 428 | aggregator='mean', concat=False, shared_aggregators=None, |