5 Best Ways to Visualize Data with TensorFlow and Python

Rate this post

💡 Problem Formulation: Data visualization is crucial for interpreting the complex relationships and patterns within data. Using TensorFlow and Python, this article aims to elucidate how data scientists and developers can visually analyze their machine learning data. In the context of a neural network training process, the desired output is visual artifacts that represent the distribution and dynamics of weights, biases, and performance metrics like loss and accuracy over time.

Method 1: TensorBoard – Visualizing Learning

The TensorBoard provides a suite of web applications designed for the visualization of TensorFlow’s computations and workflows. It’s equipped with functionalities like graph visualization, performance metrics, and histogram distributions. Users can monitor the training and validation metrics, such as loss and accuracy, in real time, providing an interactive exploration environment.

Here’s an example:

from tensorflow.keras.callbacks import TensorBoard
import time

# Name of the TensorBoard session
NAME = f"my_model-{int(time.time())}"
tensorboard = TensorBoard(log_dir=f'logs/{NAME}')

# Model fitting with TensorBoard callback
model.fit(x_train, y_train, epochs=10, validation_split=0.2, callbacks=[tensorboard])

The output of this code is the TensorBoard interface displaying various graphs and metrics when accessed via a web browser.

This code snippet initializes TensorBoard as a callback while training a TensorFlow model. The logs are saved in real-time under a directory specific to the current session, based on the time of execution, which allows easy identification and traceability of different training runs.

Method 2: Matplotlib Integration

TensorFlow can also directly integrate with Matplotlib, a popular Python plotting library, for immediate data visualization needs. The integration makes use of TensorFlow’s ability to evaluate tensors and feed the resulting data into Matplotlib’s plotting functions to create customizable static plots.

Here’s an example:

import matplotlib.pyplot as plt
import tensorflow as tf

data = tf.random.normal([100])
plt.hist(data.numpy(), bins=30)
plt.title('TensorFlow Data Histogram')

The output is a static histogram plot displaying the distribution of the TensorFlow-generated data.

The example above generates a random set of data using TensorFlow’s random normal function and then utilizes the Matplotlib library to plot a histogram. The data tensor is converted to a NumPy array to be compatible with Matplotlib’s input requirements.

Method 3: Plotly for Interactive Plots

Plotly is another library that can be combined with TensorFlow to create interactive and sophisticated visualizations. It specializes in providing a variety of graph types that can be dynamically manipulated and are web-friendly, allowing users to explore their data in greater depth.

Here’s an example:

import plotly.graph_objs as go
import tensorflow as tf

data = tf.range(10)
plot_data = [go.Scatter(x=data.numpy(), y=(data ** 2).numpy())]
fig = go.Figure(data=plot_data)

The output is an interactive line chart that can be zoomed and panned.

The code creates a simple quadratic relationship plot using TensorFlow to generate the range of numbers and Plotly to visualize them. The interactive nature of Plotly allows users to explore the data points closely, making this method highly engaging.

Method 4: Seaborn for Statistical Data Visualization

Seaborn is a powerful Python visualization library based on Matplotlib that provides an interface for drawing attractive statistical graphics. It works well with pandas DataFrames and arrays, which can be created from TensorFlow’s data structures.

Here’s an example:

import seaborn as sns
import tensorflow as tf

data = tf.random.normal([100, 2])
df = pd.DataFrame(data.numpy(), columns=['x', 'y'])

sns.jointplot(x='x', y='y', data=df, kind='scatter')

The output is a scatter plot with marginal histograms showing the distribution of the two variables.

In this snippet, TensorFlow is used to generate a dataset of two-dimensional points. Seaborn’s jointplot function then visualizes the data in a scatter plot format along with the univariate distribution of each variable on the margins.

Bonus One-Liner Method 5: TensorFlow’s Native Visualization

Lastly, TensorFlow itself provides certain native functions for quick and simple visualizations, such as image summary functions that help in viewing images during training.

Here’s an example:

from tensorflow.summary import create_file_writer
import tensorflow as tf

# Sample image tensor
image = tf.random.uniform(shape=(100, 100, 3))

# Log the image tensor
writer = create_file_writer('logs')
with writer.as_default():
    tf.summary.image("Random Image", tf.expand_dims(image, 0), step=0)

The output is a single image logged in TensorBoard’s interface under the ‘Images’ tab.

This code demonstrates how to log images directly in TensorFlow’s logging system and visualize them in TensorBoard. This is especially useful when working with image data such as results from convolutional neural networks (CNNs).


  • Method 1: TensorBoard. The strength lies in its deep integration with TensorFlow, allowing for real-time, interactive, and in-depth analysis of training processes. However, TensorBoard requires a separate web interface which might be less convenient for quick checks or when sharing results requiring viewers to have access to the logs.
  • Method 2: Matplotlib Integration. Matplotlib is well-known and has a multitude of plotting capabilities, making it suitable for static visualizations. The limitation is the loss of interactivity and sometimes complexity in creating advanced plots.
  • Method 3: Plotly for Interactive Plots. Interactive plots from Plotly are excellent for exploratory analysis and presentation-quality visuals. The possible downside is the added complexity compared to matplotlib for simple visualizations.
  • Method 4: Seaborn for Statistical Data Visualization. Seaborn offers high-level abstraction for statistical graphics, making it easier to generate more informative plots. The abstraction, however, can sometimes hide useful customizability of plots.
  • Method 5: TensorFlow’s Native Visualization. It’s quick to use within a TensorFlow script and effective for image data. The limitation is its specificity, and it’s not a general-purpose visualization tool.