FTRL Anpassung

db3b17ff · Dennis Willers · 54988faa · db3b17ff · db3b17ff
Commit db3b17ff authored Feb 19, 2023 by Dennis Willers 🏀
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

config.yaml ressources/config/config.yaml +1 -1

defineKNN.py src/knn/defineKNN.py +3 -1

No files found.
--- a/ressources/config/config.yaml
+++ b/ressources/config/config.yaml
@@ -6,7 +6,7 @@ bilder:
 knn:
  epochs: 10
  exception_optimization_method:
-    ['SGD', 'Ftrl']
+    ['SGD', 'Adam']
  exception_activation_funktion_2:
    ['ReLU']
  ignoreKnnCombinations:

--- a/src/knn/defineKNN.py
+++ b/src/knn/defineKNN.py
@@ -51,5 +51,7 @@ def get_optimization_method(optimization_method):
    if optimization_method == Optimierungsverfahren.Adam:
        return tf.keras.optimizers.Adam()
    if optimization_method == Optimierungsverfahren.Ftrl:
-        return tf.keras.optimizers.Ftrl()
+        return tf.keras.optimizers.legacy.Ftrl(learning_rate=0.001,
+                                               learning_rate_power=-0.5,
+                                               initial_accumulator_value=0.1)
    return None