Spaces:
Sleeping
Sleeping
File size: 654 Bytes
383bfb8 6f74e93 383bfb8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from .lstm import LSTM
from .gru import GRU
from .transformer import TransformerEncoder
encoder_list = ['lstm', 'gru', 'transformer']
def get_encoder(params, *args):
if not params.encoder_type in encoder_list:
raise NotImplementedError(
"Unsupported Classifier: {}".format(params.encoder_type))
if params.encoder_type == "transformer":
pass
elif params.encoder_type == "lstm":
encoder = LSTM(params, *args)
elif params.encoder_type == "gru":
encoder = GRU(params, *args)
else:
raise NotImplementedError("Unsupported Encoder: {}".format(params.encoder_type))
return encoder
|