xuyige
5 years ago
1 changed files with 4 additions and 3 deletions
Unified View
Diff Options
+ 4
- 3
fastNLP/embeddings/__init__.py
View File
| @@ -7,6 +7,7 @@ torch.FloatTensor。所有的embedding都可以使用 `self.num_embedding` 获 | |||||
| __all__ = [ | __all__ = [ | ||||
| "Embedding", | "Embedding", | ||||
| "TokenEmbedding", | |||||
| "StaticEmbedding", | "StaticEmbedding", | ||||
| "ElmoEmbedding", | "ElmoEmbedding", | ||||
| "BertEmbedding", | "BertEmbedding", | ||||
| @@ -14,14 +15,14 @@ __all__ = [ | |||||
| "StackEmbedding", | "StackEmbedding", | ||||
| "LSTMCharEmbedding", | "LSTMCharEmbedding", | ||||
| "CNNCharEmbedding", | "CNNCharEmbedding", | ||||
| "get_embeddings" | |||||
| "get_embeddings", | |||||
| ] | ] | ||||
| from .embedding import Embedding | |||||
| from .embedding import Embedding, TokenEmbedding | |||||
| from .static_embedding import StaticEmbedding | from .static_embedding import StaticEmbedding | ||||
| from .elmo_embedding import ElmoEmbedding | from .elmo_embedding import ElmoEmbedding | ||||
| from .bert_embedding import BertEmbedding, BertWordPieceEncoder | from .bert_embedding import BertEmbedding, BertWordPieceEncoder | ||||
| from .char_embedding import CNNCharEmbedding, LSTMCharEmbedding | from .char_embedding import CNNCharEmbedding, LSTMCharEmbedding | ||||
| from .stack_embedding import StackEmbedding | from .stack_embedding import StackEmbedding | ||||
| from .utils import get_embeddings | |||||
| from .utils import get_embeddings | |||||