from trains.singleTask.EF_LSTM import EF_LSTM
from trains.singleTask.LF_DNN import LF_DNN
from trains.singleTask.LMF import LMF
from trains.singleTask.TFN import TFN
from trains.singleTask.MFN import MFN
from trains.singleTask.Graph_MFN import Graph_MFN
# from trains.singleTask.BERT_MAG import BERT_MAG
from trains.singleTask.MULT import MULT
from trains.singleTask.MISA import MISA

__all__ = ['EF_LSTM', 'LF_DNN', 'LMF', 'TFN', 'MFN', 'Graph_MFN', 'MULT', 'MISA']