Text Embedding - TensorFlow, MxNet

This is a supervised text embedding algorithm which supports many pre-trained models available in MXNet and Tensorflow Hub. The following sample notebook demonstrates 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