FS-TFP/federatedscope/mf/model/__init__.py

5 lines
188 B
Python

from federatedscope.mf.model.model import BasicMFNet, VMFNet, HMFNet
from federatedscope.mf.model.model_builder import get_mfnet
__all__ = ["get_mfnet", "BasicMFNet", "VMFNet", "HMFNet"]