From da147794694e4b8ff4aa800272f2eb783dec9593 Mon Sep 17 00:00:00 2001 From: Ashutosh Hathidara Date: Thu, 12 Mar 2020 19:25:06 +0530 Subject: [PATCH] Corrected order of parameters in keras WideDeepModel --- tensorflow/python/keras/premade/wide_deep.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tensorflow/python/keras/premade/wide_deep.py b/tensorflow/python/keras/premade/wide_deep.py index 5823144d2c1..8638d3afc71 100644 --- a/tensorflow/python/keras/premade/wide_deep.py +++ b/tensorflow/python/keras/premade/wide_deep.py @@ -42,14 +42,14 @@ class WideDeepModel(keras_training.Model): linear_model = LinearModel() dnn_model = keras.Sequential([keras.layers.Dense(units=64), keras.layers.Dense(units=1)]) - combined_model = WideDeepModel(dnn_model, linear_model) + combined_model = WideDeepModel(linear_model, dnn_model) combined_model.compile(optimizer=['sgd', 'adam'], 'mse', ['mse']) # define dnn_inputs and linear_inputs as separate numpy arrays or # a single numpy array if dnn_inputs is same as linear_inputs. - combined_model.fit([dnn_inputs, linear_inputs], y, epochs) + combined_model.fit([linear_inputs, dnn_inputs], y, epochs) # or define a single `tf.data.Dataset` that contains a single tensor or # separate tensors for dnn_inputs and linear_inputs. - dataset = tf.data.Dataset.from_tensors(([dnn_inputs, linear_inputs], y)) + dataset = tf.data.Dataset.from_tensors(([linear_inputs, dnn_inputs], y)) combined_model.fit(dataset, epochs) ``` @@ -64,9 +64,9 @@ class WideDeepModel(keras_training.Model): dnn_model = keras.Sequential([keras.layers.Dense(units=1)]) dnn_model.compile('rmsprop', 'mse') dnn_model.fit(dnn_inputs, y, epochs) - combined_model = WideDeepModel(dnn_model, linear_model) + combined_model = WideDeepModel(linear_model, dnn_model) combined_model.compile(optimizer=['sgd', 'adam'], 'mse', ['mse']) - combined_model.fit([dnn_inputs, linear_inputs], y, epochs) + combined_model.fit([linear_inputs, dnn_inputs], y, epochs) ``` """