| ''' | |
| Author: Qiguang Chen | |
| Date: 2023-01-11 10:39:26 | |
| LastEditors: Qiguang Chen | |
| LastEditTime: 2023-02-18 19:33:34 | |
| Description: | |
| ''' | |
| from common.utils import InputData | |
| from model.encoder.base_encoder import BaseEncoder, BiEncoder | |
| from model.encoder.pretrained_encoder import PretrainedEncoder | |
| from model.encoder.non_pretrained_encoder import NonPretrainedEncoder | |
| class AutoEncoder(BaseEncoder): | |
| def __init__(self, **config): | |
| """automatedly load encoder by 'encoder_name' | |
| Args: | |
| config (dict): | |
| encoder_name (str): support ["lstm", "self-attention-lstm", "bi-encoder"] and other pretrained model in hugging face | |
| **args (Any): other configuration items corresponding to each module. | |
| """ | |
| super().__init__() | |
| self.config = config | |
| if config.get("encoder_name"): | |
| encoder_name = config.get("encoder_name").lower() | |
| if encoder_name in ["lstm", "self-attention-lstm"]: | |
| self.__encoder = NonPretrainedEncoder(**config) | |
| elif encoder_name == "bi-encoder": | |
| self.__encoder= BiEncoder(self.__init__(**config["intent_encoder"]), self.__init__(**config["intent_encoder"])) | |
| else: | |
| self.__encoder = PretrainedEncoder(**config) | |
| else: | |
| raise ValueError("There is no Encoder Name in config.") | |
| def forward(self, inputs: InputData): | |
| return self.__encoder(inputs) |