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

Available Models

Model ID

Fine Tunable?

Latest Version

Min SDK Version

Source

tensorflow-tc-albert-en-base

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-bert-en-cased-L-12-H-768-A-12-2

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-bert-en-cased-L-24-H-1024-A-16-2

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-bert-en-uncased-L-12-H-768-A-12-2

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-bert-en-uncased-L-24-H-1024-A-16-2

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-bert-en-wwm-cased-L-24-H-1024-A-16-2

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-bert-en-wwm-uncased-L-24-H-1024-A-16-2

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-bert-multi-cased-L-12-H-768-A-12-2

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-electra-base-1

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-electra-small-1

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-experts-bert-pubmed-1

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-experts-bert-wiki-books-1

True

2.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-10-H-128-A-2

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-10-H-256-A-4

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-10-H-512-A-8

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-10-H-768-A-12

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-12-H-128-A-2

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-12-H-256-A-4

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-12-H-512-A-8

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-12-H-768-A-12

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-2-H-128-A-2

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-2-H-256-A-4

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-2-H-512-A-8

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-2-H-768-A-12

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-4-H-128-A-2

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-4-H-256-A-4

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-4-H-512-A-8

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-4-H-768-A-12

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-6-H-128-A-2

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-6-H-256-A-4

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-6-H-512-A-8

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-6-H-768-A-12

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-8-H-128-A-2

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-8-H-256-A-4

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-8-H-512-A-8

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-small-bert-bert-en-uncased-L-8-H-768-A-12

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-talking-heads-base

True

1.0.1

2.80.0

Tensorflow Hub

tensorflow-tc-talking-heads-large

True

1.0.1

2.80.0

Tensorflow Hub