TensorFlow Hub is a repository of pre-trained TensorFlow models. In this project, we use pre-trained models from TensorFlow Hub with tf.keras for text classification. Transfer learning makes it possible to save training resources and to achieve good model generalization even when training on a small dataset. We will demonstrate this by training with several different TF-Hub modules.
-
Notifications
You must be signed in to change notification settings - Fork 0
nykolai-d/transfer-learning-nlp-tf-hub
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Using pre-trained models from TensorFlow Hub with tf.keras for text classification
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published