From 96210bf9a0fbd7615f98d915412a7430cc63e4e9 Mon Sep 17 00:00:00 2001 From: weasdown <34220924+weasdown@users.noreply.github.com> Date: Mon, 3 Feb 2025 15:54:49 +0000 Subject: [PATCH] fix: fix call to optimizers.SGD(). Also fixes test_deep_learning4e --- deep_learning4e.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deep_learning4e.py b/deep_learning4e.py index 9f5b0a8f7..7250751d3 100644 --- a/deep_learning4e.py +++ b/deep_learning4e.py @@ -575,7 +575,7 @@ def AutoencoderLearner(inputs, encoding_size, epochs=200, verbose=False): model.add(Dense(input_size, activation='relu', kernel_initializer='random_uniform', bias_initializer='ones')) # update model with sgd - sgd = optimizers.SGD(lr=0.01) + sgd = optimizers.SGD(learning_rate=0.01) model.compile(loss='mean_squared_error', optimizer=sgd, metrics=['accuracy']) # train the model