Text Classification - TensorFlow¶
This is a supervised text classification algorithm which supports fine-tuning of many pre-trained models available in Tensorflow Hub. The following sample notebook demonstrates how to use the Sagemaker Python SDK for Text Classification for using these algorithms.
For detailed documentation please refer Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK
Model ID |
Fine Tunable? |
Latest Version |
Min SDK Version |
Source |
---|---|---|---|---|
tensorflow-tc-albert-en-base |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-bert-en-cased-L-12-H-768-A-12-2 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-bert-en-cased-L-24-H-1024-A-16-2 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-bert-en-uncased-L-12-H-768-A-12-2 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-bert-en-uncased-L-24-H-1024-A-16-2 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-bert-en-wwm-cased-L-24-H-1024-A-16-2 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-bert-en-wwm-uncased-L-24-H-1024-A-16-2 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-bert-multi-cased-L-12-H-768-A-12-2 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-electra-base-1 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-electra-small-1 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-experts-bert-pubmed-1 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-experts-bert-wiki-books-1 |
True |
3.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-10-H-128-A-2 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-10-H-256-A-4 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-10-H-512-A-8 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-10-H-768-A-12 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-12-H-128-A-2 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-12-H-256-A-4 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-12-H-512-A-8 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-12-H-768-A-12 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-2-H-128-A-2 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-2-H-256-A-4 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-2-H-512-A-8 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-2-H-768-A-12 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-4-H-128-A-2 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-4-H-256-A-4 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-4-H-512-A-8 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-4-H-768-A-12 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-6-H-128-A-2 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-6-H-256-A-4 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-6-H-512-A-8 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-6-H-768-A-12 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-8-H-128-A-2 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-8-H-256-A-4 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-8-H-512-A-8 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-small-bert-bert-en-uncased-L-8-H-768-A-12 |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-talking-heads-base |
True |
2.0.1 |
2.189.0 |
|
tensorflow-tc-talking-heads-large |
True |
2.0.1 |
2.189.0 |