Decrease the learning rate for transfer learning test
This helps smooth out the learning curve, making the test be more
predictable... much less likely to fail due to a random fluctuation.
Co-authored-by: Orhan Kislal <okislal@apache.org>
diff --git a/src/ports/postgres/modules/deep_learning/test/madlib_keras_transfer_learning.sql_in b/src/ports/postgres/modules/deep_learning/test/madlib_keras_transfer_learning.sql_in
index d17ea20..92f2277 100644
--- a/src/ports/postgres/modules/deep_learning/test/madlib_keras_transfer_learning.sql_in
+++ b/src/ports/postgres/modules/deep_learning/test/madlib_keras_transfer_learning.sql_in
@@ -290,8 +290,8 @@
'mst_table',
ARRAY[1,3],
ARRAY[
- $$loss='categorical_crossentropy',optimizer='Adam(lr=0.01)',metrics=['accuracy']$$,
- $$loss='categorical_crossentropy', optimizer='Adam(lr=0.001)',metrics=['accuracy']$$
+ $$loss='categorical_crossentropy',optimizer='Adam(lr=0.00001)',metrics=['accuracy']$$,
+ $$loss='categorical_crossentropy', optimizer='Adam(lr=0.00002)',metrics=['accuracy']$$
],
ARRAY[
$$batch_size=5,epochs=1$$
