An embedding layer for fastText (:cite:`bojanowski2017enriching`). Args: key: Field name. embedding_dim: Size of this embedding layer
(self, key, embedding_dim: int)
| 61 | class FastTextEmbeddingModule(SelectFromBatchModule): |
| 62 | |
| 63 | def __init__(self, key, embedding_dim: int) -> None: |
| 64 | """An embedding layer for fastText (:cite:`bojanowski2017enriching`). |
| 65 | |
| 66 | Args: |
| 67 | key: Field name. |
| 68 | embedding_dim: Size of this embedding layer |
| 69 | """ |
| 70 | super().__init__(key) |
| 71 | self.embedding_dim = embedding_dim |
| 72 | |
| 73 | def __call__(self, batch: dict, mask=None, **kwargs): |
| 74 | outputs = super().__call__(batch, **kwargs) |