From 0ac6270f5bd1f20b64610194dada8dc503e4df0f Mon Sep 17 00:00:00 2001 From: Nils <47601016+Gradwanderer@users.noreply.github.com> Date: Mon, 13 May 2024 15:12:19 +0200 Subject: [PATCH] Update auto_encoder.py Tensorflow keras has changed names to MeanSquaredError. Please accept for the next users. --- pyod/models/auto_encoder.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pyod/models/auto_encoder.py b/pyod/models/auto_encoder.py index b330eb226..24ed1ec7a 100644 --- a/pyod/models/auto_encoder.py +++ b/pyod/models/auto_encoder.py @@ -22,12 +22,12 @@ from keras.models import Sequential from keras.layers import Dense, Dropout from keras.regularizers import l2 - from keras.losses import mean_squared_error + from keras.losses import MeanSquaredError else: from tensorflow.keras.models import Sequential from tensorflow.keras.layers import Dense, Dropout from tensorflow.keras.regularizers import l2 - from tensorflow.keras.losses import mean_squared_error + from tensorflow.keras.losses import MeanSquaredError # noinspection PyUnresolvedReferences,PyPep8Naming,PyTypeChecker @@ -135,7 +135,7 @@ class AutoEncoder(BaseDetector): def __init__(self, hidden_neurons=None, hidden_activation='relu', output_activation='sigmoid', - loss=mean_squared_error, optimizer='adam', + loss=MeanSquaredError, optimizer='adam', epochs=100, batch_size=32, dropout_rate=0.2, l2_regularizer=0.1, validation_size=0.1, preprocessing=True, verbose=1, random_state=None, contamination=0.1):