使用 [4, 4] 个隐藏层在 iris 数据集上训练并在 tensorflow 和 keras 中分别创建的神经网络给出了不同的结果。
虽然 tensorflow 模型在测试中给出了 96.6% 的准确率,但 keras 模型只给出了大约 50%。各种超参数,如学习率、优化器、小批量大小等,在两种情况下都是相同的。
Keras 模型
model = Sequential()
model.add(Dense(units = 4, activation = 'relu', input_dim = 4))
model.add(Dropout(0.25))
model.add(Dense(units = 4, activation = 'relu'))
model.add(Dropout(0.25))
model.add(Dense(units = 3, activation = 'softmax'))
adam = Adam(epsilon = 10**(-6), lr = 0.01)
model.compile(optimizer = 'adagrad', loss = 'categorical_crossentropy', metrics = ['accuracy'])
one_hot_labels = keras.utils.to_categorical(y_train, num_classes = 3)
model.fit(X_train, one_hot_labels, epochs = 50, batch_size = 40)
Tensorflow 模型
feature_columns = [tf.feature_column.numeric_column(key = name,
shape = (1),
dtype = tf.float32) for name in list(X_train.columns)]
classifier = tf.estimator.DNNClassifier(hidden_units = [4, 4],
feature_columns = feature_columns,
n_classes = 3,
dropout = 0.25,
model_dir = './DNN_model')
train_input_fn = tf.estimator.inputs.pandas_input_fn(x = X_train,
y = y_train,
batch_size = 40,
num_epochs = 50,
shuffle = False)
classifier.train(input_fn = train_input_fn, steps = None)
对于 keras 模型,我确实尝试过改变学习率、增加 epoch 的数量、使用不同的优化器等。因此,准确性仍然很差。显然,这两个模型都在做不同的事情,但从表面上看,它们在所有关键方面对我来说似乎都是一样的。
感谢任何帮助。
请您参考如下方法:
它们具有相同的架构,仅此而已。
性能差异来自以下一个或多个因素:
你有辍学。因此,您的网络在每次开始时都会表现不同(检查 Dropout 的工作原理);
权重初始化,您在 Keras 和 TensorFlow 中使用哪种方法?
检查优化器的所有参数。
