Skip to Content
Mastering TensorFlow 1.x
book

Mastering TensorFlow 1.x

by Armando Fandango
January 2018
Intermediate to advanced
474 pages
10h 23m
English
Packt Publishing
Content preview from Mastering TensorFlow 1.x

GRU in TensorFlow

To change the LSTM example in the last section to the GRU network, change the cell type as follows and the TensorFlow takes care of the rest for you:

cell = tf.nn.rnn_cell.GRUCell(state_size)

The complete code for the GRU model is provided in the notebook ch-07a_RNN_TimeSeries_TensorFlow.

For the small airpass dataset, the GRU has shown better performance for the same number of epochs. In practice, GRU and LSTM have shown comparable performance. In terms of execution speed, the GRU model trains and predicts faster as compared to the LSTM.

The complete code for the GRU model is provided in the Jupyter notebook. The results from the GRU model are as follows:

train mse = 0.0019633215852081776 test mse = 0.014307591132819653 ...
Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.

Read now

Unlock full access

More than 5,000 organizations count on O’Reilly

AirBnbBlueOriginElectronic ArtsHomeDepotNasdaqRakutenTata Consultancy Services

QuotationMarkO’Reilly covers everything we've got, with content to help us build a world-class technology community, upgrade the capabilities and competencies of our teams, and improve overall team performance as well as their engagement.
Julian F.
Head of Cybersecurity
QuotationMarkI wanted to learn C and C++, but it didn't click for me until I picked up an O'Reilly book. When I went on the O’Reilly platform, I was astonished to find all the books there, plus live events and sandboxes so you could play around with the technology.
Addison B.
Field Engineer
QuotationMarkI’ve been on the O’Reilly platform for more than eight years. I use a couple of learning platforms, but I'm on O'Reilly more than anybody else. When you're there, you start learning. I'm never disappointed.
Amir M.
Data Platform Tech Lead
QuotationMarkI'm always learning. So when I got on to O'Reilly, I was like a kid in a candy store. There are playlists. There are answers. There's on-demand training. It's worth its weight in gold, in terms of what it allows me to do.
Mark W.
Embedded Software Engineer

You might also like

TensorFlow 1.x Deep Learning Cookbook

TensorFlow 1.x Deep Learning Cookbook

Antonio Gulli, Dr. Amita Kapoor
What's New in TensorFlow 2.0

What's New in TensorFlow 2.0

Ajay Baranwal, Alizishaan Khatri, Tanish Baranwal
Machine Learning with TensorFlow 1.x

Machine Learning with TensorFlow 1.x

Quan Hua, Saif Ahmed, Shams Ul Azeem

Publisher Resources

ISBN: 9781788292061