azure - 未知优化器 : Ftrl on azure databricks tf keras but other optimizers work just fine

Question

我构建了一个简单的人工神经网络，对数据上不同优化器的性能进行一些简单的前期比较。代码片段如下。我只是使用默认参数遍历不同的优化器，并比较一些基本指标的性能。我在 Azure Databricks 上的仅 CPU 集群上运行 tf keras，如下所示:keras 版本:2.2.4-tf； tensorflow 版本:1.15.0

所有 keras 优化器 ( https://keras.io/api/optimizers/ ) 在我的代码中都工作得很好，但 Ftrl 除外，我收到错误:

ValueError:未知优化器:Ftrl

我尝试过“Ftrl”、“ftrl”并将优化器设置为变量并调用它。没有骰子。

感谢您的帮助

import tensorflow as tf
from tensorflow import keras
from keras.layers import Dense
from keras.models import Sequential

model = Sequential()
model.add(Dense(300, activation = 'relu', input_shape=(161,)))
model.add(Dense(300, activation = 'relu'))
model.add(Dense(300, activation = 'relu'))
model.add(Dense(2, activation = 'softmax'))

model.compile(loss="categorical_crossentropy",
             optimizer = "Ftrl",
             metrics=['accuracy'])

Answer 1

您可以简单地使用 tf.keras.optimizers.Ftrl()

model.compile(loss="categorical_crossentropy",
         optimizer = tf.keras.optimizers.Ftrl(),
         metrics=['accuracy'])