【发布时间】: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