FateWithShellmiaoComment/Fate/python/federatedml/nn/__init__.py

from federatedml.nn.dataset.base import Dataset
from federatedml.nn.homo.trainer.trainer_base import TrainerBase

__all__ = ['Dataset', 'TrainerBase']