【问题标题】:Unknown optimizer: Ftrl on azure databricks tf keras but other optimizers work just fine未知优化器:天蓝色数据块上的 Ftrl tf keras 但其他优化器工作得很好
【发布时间】:2020-10-20 05:16:26
【问题描述】:

我已经构建了一个简单的人工神经网络,用于对我的数据上的不同优化器性能进行一些简单的前期比较。代码 sn-p 如下。我只是在使用默认参数进行不同的优化器并比较一些基本措施的性能。 我在 Azure Databricks 上的仅 CPU 群集上运行 tf keras,如下所示: keras 版本:2.2.4-tf; 张量流版本: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'])

【问题讨论】:

    标签: azure tensorflow keras databricks


    【解决方案1】:

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

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

    【讨论】:

      猜你喜欢
      • 2020-11-09
      • 2020-11-15
      • 2014-04-18
      • 1970-01-01
      • 2020-03-14
      • 2020-11-07
      • 1970-01-01
      • 2023-01-30
      • 1970-01-01
      相关资源
      最近更新 更多