Skip to content

Evaluation künstlicher neuronaler Netze für eine OOS-Erkennung
Evaluation künstlicher neuronaler Netze für eine OOS-Erkennung
Commit db3b17ff authored by Dennis Willers's avatar Dennis Willers 🏀
Browse files
Options

FTRL Anpassung

parent 54988faa
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
ressources/config/config.yaml
View file @ db3b17ff
...@@ -6,7 +6,7 @@ bilder: ...@@ -6,7 +6,7 @@ bilder:
knn: knn:
epochs: 10 epochs: 10
exception_optimization_method: exception_optimization_method:
['SGD', 'Ftrl'] ['SGD', 'Adam']
exception_activation_funktion_2: exception_activation_funktion_2:
['ReLU'] ['ReLU']
ignoreKnnCombinations: ignoreKnnCombinations:
......
src/knn/defineKNN.py
View file @ db3b17ff
...@@ -51,5 +51,7 @@ def get_optimization_method(optimization_method): ...@@ -51,5 +51,7 @@ def get_optimization_method(optimization_method):
if optimization_method == Optimierungsverfahren.Adam: if optimization_method == Optimierungsverfahren.Adam:
return tf.keras.optimizers.Adam() return tf.keras.optimizers.Adam()
if optimization_method == Optimierungsverfahren.Ftrl: 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 return None
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment