Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge

tfmot.clustering.keras.ClusteringSummaries

Helper class to create tensorboard summaries for the clustering progress.

This class is derived from tf.keras.callbacks.TensorBoard and just adds functionality to write histograms with batch-wise frequency.

log_dir The path to the directory where the log files are saved
cluster_update_freq determines the frequency of updates of the clustering histograms. Same behaviour as parameter update_freq of the base class, i.e. it accepts 'batch', 'epoch' or integer.

Methods

set_model

Sets Keras model and writes graph if specified.

set_params