Skip to main content
 首页 » 编程设计

python之相同的 NN 架构在 tensorflow 和 keras 中给出不同的精度

2024年08月15日31bluestorm

使用 [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 中使用哪种方法?

  • 检查优化器的所有参数。