Spaces:
Runtime error
Runtime error
File size: 1,019 Bytes
7900c16 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from tencentpretrain.encoders.transformer_encoder import TransformerEncoder
from tencentpretrain.encoders.rnn_encoder import RnnEncoder
from tencentpretrain.encoders.rnn_encoder import LstmEncoder
from tencentpretrain.encoders.rnn_encoder import GruEncoder
from tencentpretrain.encoders.rnn_encoder import BirnnEncoder
from tencentpretrain.encoders.rnn_encoder import BilstmEncoder
from tencentpretrain.encoders.rnn_encoder import BigruEncoder
from tencentpretrain.encoders.cnn_encoder import GatedcnnEncoder
from tencentpretrain.encoders.dual_encoder import DualEncoder
str2encoder = {"transformer": TransformerEncoder, "rnn": RnnEncoder, "lstm": LstmEncoder,
"gru": GruEncoder, "birnn": BirnnEncoder, "bilstm": BilstmEncoder, "bigru": BigruEncoder,
"gatedcnn": GatedcnnEncoder, "dual": DualEncoder}
__all__ = ["TransformerEncoder", "RnnEncoder", "LstmEncoder", "GruEncoder", "BirnnEncoder",
"BilstmEncoder", "BigruEncoder", "GatedcnnEncoder", "DualEncoder", "str2encoder"]
|