diff --git a/pyod/models/vae.py b/pyod/models/vae.py index 4938cbc8c..a1ea83e63 100644 --- a/pyod/models/vae.py +++ b/pyod/models/vae.py @@ -7,7 +7,7 @@ :cite:`kingma2013auto` Kingma, Diederik, Welling 'Auto-Encodeing Variational Bayes' https://arxiv.org/abs/1312.6114 - + :cite:`burgess2018understanding` Burges et al 'Understanding disentangling in beta-VAE' https://arxiv.org/pdf/1804.03599.pdf @@ -221,7 +221,7 @@ def sampling(self, args): ---------- args : tensor Mean and log of variance of Q(z|X). - + Returns ------- z : tensor @@ -349,7 +349,7 @@ def fit(self, X, y=None): else: X_norm = np.copy(X) - pred_scores = self.model_.predict(X_norm) + pred_scores = self.model_.predict(X_norm, verbose=self.verbose) self.decision_scores_ = pairwise_distances_no_broadcast(X_norm, pred_scores) self._process_decision_scores() @@ -382,5 +382,5 @@ def decision_function(self, X): X_norm = np.copy(X) # Predict on X and return the reconstruction errors - pred_scores = self.model_.predict(X_norm) + pred_scores = self.model_.predict(X_norm, verbose=self.verbose) return pairwise_distances_no_broadcast(X_norm, pred_scores)