Text Embedding - TensorFlow, MxNet — sagemaker 2.199.0 documentation (original) (raw)
This is a supervised text embedding algorithm which supports many pre-trained models available in MXNet and Tensorflow Hub. The followingsample notebookdemonstrates how to use the Sagemaker Python SDK for Text Embedding for using these algorithms.
For detailed documentation please refer Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK
Available Models¶
| Model ID | Fine Tunable? | Latest Version | Min SDK Version | Source |
|---|---|---|---|---|
| mxnet-tcembedding-robertafin-base-uncased | False | 2.0.0 | 2.189.0 | GluonCV |
| mxnet-tcembedding-robertafin-base-wiki-uncased | False | 2.0.0 | 2.189.0 | GluonCV |
| mxnet-tcembedding-robertafin-large-uncased | False | 2.0.0 | 2.189.0 | GluonCV |
| mxnet-tcembedding-robertafin-large-wiki-uncased | False | 2.0.0 | 2.189.0 | GluonCV |
| tensorflow-tcembedding-bert-en-uncased-L-10-H-128-A-2-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-10-H-256-A-4-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-10-H-512-A-8-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-10-H-768-A-12-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-12-H-128-A-2-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-12-H-256-A-4 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-12-H-512-A-8-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-12-H-768-A-12-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-12-H-768-A-12-4 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-2-H-128-A-2-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-2-H-256-A-4 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-2-H-512-A-8-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-2-H-768-A-12-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-4-H-128-A-2-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-4-H-256-A-4-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-4-H-512-A-8-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-4-H-768-A-12-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-6-H-128-A-2-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-6-H-256-A-4 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-6-H-512-A-8-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-6-H-768-A-12-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-8-H-256-A-4-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-8-H-512-A-8-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-en-uncased-L-8-H-768-A-12-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-wiki-books-mnli-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-bert-wiki-books-sst2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-talkheads-ggelu-bert-en-base-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-talkheads-ggelu-bert-en-large-2 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-universal-sentence-encoder-cmlm-en-base-1 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |
| tensorflow-tcembedding-universal-sentence-encoder-cmlm-en-large-1 | False | 2.0.0 | 2.189.0 | Tensorflow Hub |