未知的优化器: azure databricks tf keras上的Ftrl，但其他优化器工作正常

Question

我已经构建了一个简单的ANN来对我的数据进行一些简单的、预先比较不同优化器性能的工作。下面是代码片段。我只是在使用默认参数浏览不同的优化器，并比较一些基本度量的性能。我在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'])