In [1]:
#@title Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# See the License for the specific language governing permissions and
# limitations under the License.

TensorFlow 2 quickstart for beginners

This short introduction uses Keras to:

  1. Build a neural network that classifies images.
  2. Train this neural network.
  3. And, finally, evaluate the accuracy of the model.

This is a Google Colaboratory notebook file. Python programs are run directly in the browser—a great way to learn and use TensorFlow. To follow this tutorial, run the notebook in Google Colab by clicking the button at the top of this page.

  1. In Colab, connect to a Python runtime: At the top-right of the menu bar, select CONNECT.
  2. Run all the notebook code cells: Select Runtime > Run all.

Download and install the TensorFlow 2 package. Import TensorFlow into your program:

In [1]:
from __future__ import absolute_import, division, print_function, unicode_literals

# Install TensorFlow
  # %tensorflow_version only exists in Colab.
  %tensorflow_version 2.x
except Exception:

import tensorflow as tf

Load and prepare the MNIST dataset. Convert the samples from integers to floating-point numbers:

In [2]:
mnist = tf.keras.datasets.mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0
Downloading data from
11493376/11490434 [==============================] - 0s 0us/step

Build the tf.keras.Sequential model by stacking layers. Choose an optimizer and loss function for training:

In [3]:
model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(128, activation='relu'),
  tf.keras.layers.Dense(10, activation='softmax')

WARNING:tensorflow:From D:\Data\PredictiveModeler\Anaconda\envs\tensorflow_env\lib\site-packages\tensorflow\python\ops\ calling VarianceScaling.__init__ (from tensorflow.python.ops.init_ops) with dtype is deprecated and will be removed in a future version.
Instructions for updating:
Call initializer instance with the dtype argument instead of passing it to the constructor

Train and evaluate the model:

In [4]:, y_train, epochs=5)

model.evaluate(x_test,  y_test, verbose=2)
Epoch 1/5
60000/60000 [==============================] - 22s 360us/sample - loss: 0.2902 - acc: 0.9158
Epoch 2/5
60000/60000 [==============================] - 23s 376us/sample - loss: 0.1405 - acc: 0.9588
Epoch 3/5
60000/60000 [==============================] - 22s 361us/sample - loss: 0.1045 - acc: 0.9679
Epoch 4/5
60000/60000 [==============================] - 23s 380us/sample - loss: 0.0877 - acc: 0.9727
Epoch 5/5
60000/60000 [==============================] - 21s 345us/sample - loss: 0.0731 - acc: 0.9769
10000/10000 - 0s - loss: 0.0733 - acc: 0.9784
[0.07331321267643943, 0.9784]

The image classifier is now trained to ~98% accuracy on this dataset. To learn more, read the TensorFlow tutorials.