--- a +++ b/layers/__init__.py @@ -0,0 +1,9 @@ +# -*- coding: utf-8 -*- + +from .aggregator import SumAggregator, ConcatAggregator, NeighAggregator + +Aggregator = { + 'sum': SumAggregator, + 'concat': ConcatAggregator, + 'neigh': NeighAggregator +}