Sentence Pair Classification - TensorFlow

This is a supervised sentence pair 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 Sentence Pair 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-spc-bert-en-cased-L-12-H-768-A-12-2

True

1.2.3

2.75.0

Tensorflow Hub

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

True

1.2.3

2.75.0

Tensorflow Hub

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

True

1.2.3

2.75.0

Tensorflow Hub

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

True

1.2.3

2.75.0

Tensorflow Hub

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

True

1.2.3

2.75.0

Tensorflow Hub

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

True

1.2.3

2.75.0

Tensorflow Hub

tensorflow-spc-electra-base-1

True

1.2.3

2.75.0

Tensorflow Hub

tensorflow-spc-electra-small-1

True

1.2.3

2.75.0

Tensorflow Hub

tensorflow-spc-experts-bert-pubmed-1

True

1.2.3

2.75.0

Tensorflow Hub

tensorflow-spc-experts-bert-wiki-books-1

True

1.2.3

2.75.0

Tensorflow Hub