def fit(self) -> None:
        """Fit KDE model to collected embeddings from the training set."""
        embeddings = torch.vstack(self.embeddings)

        logger.info("Fitting a KDE model to the embedding collected from the training set.")
        self.model.classifier.fit(embeddings)
Fit KDE model to collected embeddings from the training set.
src/anomalib/models/image/dfkde/lightning_model.py