20 lines
821 B
Python
20 lines
821 B
Python
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
from __future__ import division
|
|
|
|
from federatedscope.core.mlp import MLP
|
|
from federatedscope.gfl.model.model_builder import get_gnn
|
|
from federatedscope.gfl.model.gcn import GCN_Net
|
|
from federatedscope.gfl.model.sage import SAGE_Net
|
|
from federatedscope.gfl.model.gin import GIN_Net
|
|
from federatedscope.gfl.model.gat import GAT_Net
|
|
from federatedscope.gfl.model.gpr import GPR_Net
|
|
from federatedscope.gfl.model.graph_level import GNN_Net_Graph
|
|
from federatedscope.gfl.model.link_level import GNN_Net_Link
|
|
from federatedscope.gfl.model.fedsageplus import LocalSage_Plus, FedSage_Plus
|
|
|
|
__all__ = [
|
|
'get_gnn', 'GCN_Net', 'SAGE_Net', 'GIN_Net', 'GAT_Net', 'GPR_Net',
|
|
'GNN_Net_Graph', 'GNN_Net_Link', 'LocalSage_Plus', 'FedSage_Plus', 'MLP'
|
|
]
|