training_tf_data.py

model.compile(
  optimizer=Adam(),
  loss='mean_squared_error'
)

epochs = 25
log_dir='logs'

model.fit(
  train_data, # train dataset that yields (x, y) tuples
  epochs=epochs,
  validation_data=validation_data, # validation dataset that yields (x, y) tuples
  callbacks=[
    TensorBoard(log_dir=log_dir, histogram_freq=1), # log metrics
    KerasCallback(logdir, hparams) # log hparams
  ]
)