Structure a Keras Tensorboard graph

Viewed 5994

When I create a simple Keras Model

model = Sequential()
model.add(Dense(10, activation='tanh', input_dim=1))
model.add(Dense(1, activation='linear'))
model.compile(loss='mean_squared_error', optimizer='adam', metrics=['mean_squared_error'])

and do a callback to Tensorboard

tensorboard = TensorBoard(log_dir='c:/temp/tensorboard/run1', histogram_freq=1, write_graph=True, write_images=False)
model.fit(x, y, epochs=1000, batch_size=1, callbacks=[tensorboard])

The output in Tensorboard looks like this: enter image description here

In other words, it's a complete mess.

  1. Is there anything I can do to make the graphs output look more structured?
  2. How can I create histograms of weights with Keras and Tensorboard?
1 Answers
Related