#from .nlp.BasicBert import BasicBert
#from .nlp.BertPoint import BertPoint
#from .nlp.BertPoolOut import BertPoolOut
from .nlp.BertPoolOutMax import BertPoolOutMax
from .nlp.AttenRNN import AttentionRNN

model_list = {
#    "BasicBert": BasicBert,
#    "BertPoint": BertPoint,
#    "BertPoolOut": BertPoolOut,
    "BertPoolOutMax": BertPoolOutMax,
    "AttenRNN": AttentionRNN
}


def get_model(model_name):
    if model_name in model_list.keys():
        return model_list[model_name]
    else:
        raise NotImplementedError

240d828ba49f769c03a109b6c8def1dcb667e769
<mask token>

<mask token>


def get_model(model_name):
    if model_name in model_list.keys():
        return model_list[model_name]
    else:
        raise NotImplementedError

<mask token>
model_list = {'BertPoolOutMax': BertPoolOutMax, 'AttenRNN': AttentionRNN}


def get_model(model_name):
    if model_name in model_list.keys():
        return model_list[model_name]
    else:
        raise NotImplementedError

from .nlp.BertPoolOutMax import BertPoolOutMax
from .nlp.AttenRNN import AttentionRNN
model_list = {'BertPoolOutMax': BertPoolOutMax, 'AttenRNN': AttentionRNN}


def get_model(model_name):
    if model_name in model_list.keys():
        return model_list[model_name]
    else:
        raise NotImplementedError

#from .nlp.BasicBert import BasicBert
#from .nlp.BertPoint import BertPoint
#from .nlp.BertPoolOut import BertPoolOut
from .nlp.BertPoolOutMax import BertPoolOutMax
from .nlp.AttenRNN import AttentionRNN

model_list = {
#    "BasicBert": BasicBert,
#    "BertPoint": BertPoint,
#    "BertPoolOut": BertPoolOut,
    "BertPoolOutMax": BertPoolOutMax,
    "AttenRNN": AttentionRNN
}


def get_model(model_name):
    if model_name in model_list.keys():
        return model_list[model_name]
    else:
        raise NotImplementedError
