3 Ways to Easily Visualize Keras Machine Learning Models

This guide describes 3 easy ways to visualize Keras models. It explains the use of 3 specific tools: Netron, visualkeras, and TensorBoard. Tensorboard and visualkeras can be used to visualize untrained models.

What Tools Can I Use to Visualize Keras Models?

The old cliche "a picture is worth a thousand words" might be even more true when working with complex machine learning models. Unless you are blessed with a photographic memory, you can quickly lose track of the model architecture when just reading through code.

Luckily, there are some easy ways to visualize machine learning models. This guide focuses on the visualization of Keras models and it uses the following model (the "test model") for demonstration:

def build_model(pad_len, imu_dim, tof_dim, n_classes):
   def time_sum(x):
      return K.sum(x, axis=1)

   def squeeze_last_axis(x):
      return tf.squeeze(x, axis=-1)

   def expand_last_axis ...