An example of transfer learning using TensorFlow and Keras
For this task, we use a pre-trained model from google: gnews-swivel-20dim The dataset is: The Internet Movie DataBase (IMDB) reviews data-set
For each review, the pre-trained model returns a 20-dimensional vector,
We attach 2 layers to the pre-trained model:
- A 16 neuron relu layer to learn new data
- A 1 neuron softmax output layer to classify each review into +ve/-ve class
After finetuning the new model, it can classify movie reviews into +ve/-ve classes, The loss and accuracy scores are reported in the Jupyter Notebook.