【发布时间】:2019-09-27 21:06:44
【问题描述】:
我的 Tensorflow 模型出现问题,因此决定尝试 Keras。至少在我看来,我正在创建具有相同参数的相同模型,但 Tensorflow 模型只输出 train_y 的平均值,而 Keras 模型实际上根据输入而变化。我在我的 tf.Session 中遗漏了什么吗?我通常使用 Tensorflow,从来没有遇到过这样的问题。 张量流代码:
score_inputs = tf.placeholder(np.float32, shape=(None, 100))
targets = tf.placeholder(np.float32, shape=(None), name="targets")
l2 = tf.contrib.layers.l2_regularizer(0.01)
first_layer = tf.layers.dense(score_inputs, 100, activation=tf.nn.relu, kernel_regularizer=l2)
outputs = tf.layers.dense(first_layer, 1, activation = None, kernel_regularizer=l2)
optimizer = tf.train.AdamOptimizer(0.001)
l2_loss = tf.losses.get_regularization_loss()
loss = tf.reduce_mean(tf.square(tf.subtract(targets, outputs)))
loss += l2_loss
rmse = tf.sqrt(tf.reduce_mean(tf.square(outputs - targets)))
mae = tf.reduce_mean(tf.sqrt(tf.square(outputs - targets)))
training_op = optimizer.minimize(loss)
batch_size = 32
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
for epoch in range(10):
avg_train_error = []
for i in range(len(train_x) // batch_size):
batch_x = train_x[i*batch_size: (i+1)*batch_size]
batch_y = train_y[i*batch_size: (i+1)*batch_size]
_, train_loss = sess.run([training_op, loss], {score_inputs: batch_x, targets: batch_y})
feed = {score_inputs: test_x, targets: test_y}
test_loss, test_mae, test_rmse, test_ouputs = sess.run([loss, mae, rmse, outputs], feed)
平均绝对误差为 0.682,均方根误差为 0.891。
Keras 代码:
inputs = Input(shape=(100,))
hidden = Dense(100, activation="relu", kernel_regularizer = regularizers.l2(0.01))(inputs)
outputs = Dense(1, activation=None, kernel_regularizer = regularizers.l2(0.01))(hidden)
model = Model(inputs=inputs, outputs=outputs)
model.compile(optimizer=keras.optimizers.Adam(lr=0.001), loss='mse', metrics=['mae'])
model.fit(train_x, train_y, batch_size=32, epochs=10, shuffle=False)
keras_pred = model.predict(test_x)
平均绝对误差为 0.601,均方根误差为 0.753。
在我看来,我在两个实例中定义了相同的网络,但正如我所说,Tensorflow 模型只输出 train_y 的平均值,而 Keras 模型的性能要好得多。有什么建议吗?
【问题讨论】:
-
这个分数是通过测试预测的吗?
-
是的,如果我不清楚,对不起。
标签: python tensorflow keras deep-learning regression