Google I/O returns May 18-20! Reserve space and build your schedule Register now

Module: tfmot.clustering.keras

Module containing clustering code built on Keras abstractions.


class CentroidInitialization: Specifies how the cluster centroids should be initialized.


cluster_scope(...): Provides a scope in which Clustered layers and models can be deserialized.

cluster_weights(...): Modify a keras layer or model to be clustered during training.

strip_clustering(...): Strip clustering wrappers from the model.