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