[PYTHON] From running MINST on TensorFlow 2.0 to visualization on TensorBoard (2019 edition)
Introduction
- The official TensorFlow tutorial Beginner quickstart provides sample learning and evaluation using the MNIST dataset.
- In this article, I will try to actually try learning and evaluating MINST, and I will write about visualizing the learning process and data flow with TensorBoard.
environment
- Windows10 Pro 64bit
- Python 3.7.4
- Visual Studio Code 1.40
- Proxy authentication required for external connection
Introduction of TensorFlow
- Install with pip from the command prompt, and the proxy settings are also set in advance because the implementation environment is under the proxy.
set HTTP_PROXY=Proxy settings
set HTTPS_PROXY=Proxy settings
pip install tensorflow
Run MINST
- Copy the contents of the official tutorial as it is
from __future__ import absolute_import, division, print_function, unicode_literals
import tensorflow as tf
#Data load
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
#Stacking layers tf.keras.Build a Sequential model
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)), #Smoothing 28 x 28 size 2D data to 784 1D data
tf.keras.layers.Dense(128, activation='relu'), #Specify the ramp function as the activation function
tf.keras.layers.Dropout(0.2), #Specifying dropout rate to prevent overfitting
tf.keras.layers.Dense(10, activation='softmax') #Specify soft plus for activation function
])
#Choose an optimizer and loss function for training
model.compile(optimizer='adam', #Decide how to optimize learning
loss='sparse_categorical_crossentropy', #Decide how to define loss
metrics=['accuracy'])
#Train the model
model.fit(x_train, y_train, epochs=5)
#Return the loss value and evaluation value of the model
model.evaluate(x_test, y_test, verbose=2)
- If you execute it as it is, a proxy authentication error will occur in the data load part.
- Since the proxy settings are troublesome, download the data directly using a browser (Download URL is here)
- If you place the downloaded data file (mnist.npz) below, it will be read locally.
Data placement destination
C:\Users\User name\.keras\datasets
- If you execute the above script with the data file placed locally, the following result will be output.
Execution result
Train on 60000 samples
Epoch 1/5
2019-11-13 14:08:58.762382: I tensorflow/core/profiler/lib/profiler_session.cc:184] Profiler session started.
60000/60000 [==============================] - 5s 82us/sample - loss: 0.2946 - accuracy: 0.9146 - val_loss: 0.1343 - val_accuracy: 0.9589
Epoch 2/5
60000/60000 [==============================] - 4s 66us/sample - loss: 0.1435 - accuracy: 0.9578 - val_loss: 0.1014 - val_accuracy: 0.9698
Epoch 3/5
60000/60000 [==============================] - 4s 67us/sample - loss: 0.1082 - accuracy: 0.9675 - val_loss: 0.0840 - val_accuracy: 0.9737
Epoch 4/5
60000/60000 [==============================] - 4s 67us/sample - loss: 0.0852 - accuracy: 0.9743 - val_loss: 0.0780 - val_accuracy: 0.9758
Epoch 5/5
60000/60000 [==============================] - 4s 66us/sample - loss: 0.0742 - accuracy: 0.9767 - val_loss: 0.0686 - val_accuracy: 0.9780
10000/1 - 0s - loss: 0.0365 - accuracy: 0.9780
- ** accuracy: 0.9780 ** is the correct answer rate using test data
Visualization of learning process using TensorBoard
- Changed the script to output the log for TensorBoard
from __future__ import absolute_import, division, print_function, unicode_literals
import tensorflow as tf
#Data load
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
#Stacking layers tf.keras.Build a Sequential model
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)), #Smoothing 28 x 28 size 2D data to 784 1D data
tf.keras.layers.Dense(128, activation='relu'), #Specify the ramp function as the activation function
tf.keras.layers.Dropout(0.2), #Specifying dropout rate to prevent overfitting
tf.keras.layers.Dense(10, activation='softmax') #Specify soft plus for activation function
])
#Choose an optimizer and loss function for training
model.compile(optimizer='adam', #Decide how to optimize learning
loss='sparse_categorical_crossentropy', #Decide how to define loss
metrics=['accuracy'])
#Log output callback function for tensorboard
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir="log_dir", histogram_freq=1)
#Train the model
model.fit(x_train, y_train, epochs=5, validation_data=(x_test, y_test), callbacks=[tensorboard_callback])
#Return the loss value and evaluation value of the model
model.evaluate(x_test, y_test, verbose=2)
- Excerpt only for changes.
#Log output callback function for tensorboard
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir="log_dir", histogram_freq=1)
#Train the model
model.fit(x_train, y_train, epochs=5, validation_data=(x_test, y_test), callbacks=[tensorboard_callback])
- When you execute the modified script, a "log_dir" folder is created in the folder where the script is stored.
- With the "log_dir" folder existing, execute the following from the command prompt
tensorboard --logdir='./log_dir'
- You can display the TensorBoard screen by accessing the following with a browser.
- http://localhost:6006/
- Main Graph part visualizes the learning process
Summary
- I just did the contents of the official tutorial, but somehow I felt like an engineer doing machine learning.
- I don't know the details at all, so I will study little by little from now on.
Recommended Posts