View source on GitHub
|
Module containing clustering code built on Keras abstractions.
Classes
class CentroidInitialization: Specifies how the cluster centroids should be initialized.
Functions
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.