Commit 54988faa authored by Dennis Willers's avatar Dennis Willers 🏀

Learning Rate entfernt

parent 27b4eb58
......@@ -6,29 +6,11 @@ bilder:
knn:
epochs: 10
exception_optimization_method:
['SGD']
['SGD', 'Ftrl']
exception_activation_funktion_2:
['ReLU']
ignoreKnnCombinations:
[
['SGD','ReLU','sigmoid'],
['SGD','sigmoid','sigmoid'],
# Adam Complete List
['Adam','ReLU','sigmoid'],
['Adam','sigmoid','sigmoid'],
# master_1
['Adam','tanh','sigmoid'],
['Adam','softmax','sigmoid'],
['Adam','sigmoid','tanh' ],
['Adam','softmax','tanh' ],
['Adam','ReLU','tanh' ],
# master_2
['Adam','tanh','tanh'],
['Adam','sigmoid','softmax'],
['Adam','tanh','softmax'],
['Adam','ReLU','softmax'],
['Adam','softmax','softmax']
]
[]
result:
plot_path: "ressources/results/plot/"
excel_path: "ressources/results/excel/"
......@@ -49,7 +49,7 @@ def get_optimization_method(optimization_method):
if optimization_method == Optimierungsverfahren.SGD:
return tf.keras.optimizers.SGD(learning_rate=0.001, momentum=0.9)
if optimization_method == Optimierungsverfahren.Adam:
return tf.keras.optimizers.Adam(learning_rate=0.001)
return tf.keras.optimizers.Adam()
if optimization_method == Optimierungsverfahren.Ftrl:
return tf.keras.optimizers.Ftrl(learning_rate=0.001)
return tf.keras.optimizers.Ftrl()
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