| from typing import List | |
| import os | |
| import fasttext.util | |
| class PreTrainedPipeline(): | |
| def __init__(self, path=""): | |
| """ | |
| Initialize model | |
| """ | |
| self.model = fasttext.load_model(os.path.join(path, 'debate2vec.bin')) | |
| def __call__(self, inputs: str) -> List[float]: | |
| """ | |
| Args: | |
| inputs (:obj:`str`): | |
| a string to get the features of. | |
| Return: | |
| A :obj:`list` of floats: The features computed by the model. | |
| """ | |
| return self.model.get_sentence_vector(inputs).tolist() | |