Skip to content

Commit efcbf69

Browse files
committed
make run on GPU
1 parent 4e03e0a commit efcbf69

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

chapter10/RNN.ipynb

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -92,13 +92,17 @@
9292
"callback = tf.keras.callbacks.EarlyStopping(monitor='loss', patience=3)\n",
9393
"\n",
9494
"def create_model(passengers):\n",
95-
" input_layer = Input(shape=(LOOKBACK, 1))\n",
96-
" recurrent = Bidirectional(LSTM(20, activation=\"relu\"))(input_layer)\n",
97-
" output_layer = Dense(1)(recurrent)\n",
98-
" model = keras.models.Model(inputs=input_layer, outputs=output_layer)\n",
99-
" model.compile(loss='mse', optimizer=keras.optimizers.Adagrad(),\n",
100-
" metrics=[keras.metrics.RootMeanSquaredError(), keras.metrics.MeanAbsoluteError()])\n",
101-
" return model\n",
95+
" input_layer = Input(shape=(LOOKBACK, 1))\n",
96+
" recurrent = Bidirectional(LSTM(20, activation=\"tanh\"))(input_layer)\n",
97+
" output_layer = Dense(1)(recurrent)\n",
98+
" model = keras.models.Model(inputs=input_layer, outputs=output_layer)\n",
99+
" model.compile(\n",
100+
" loss='mse', optimizer=keras.optimizers.Adagrad(),\n",
101+
" metrics=[\n",
102+
" keras.metrics.RootMeanSquaredError(),\n",
103+
" keras.metrics.MeanAbsoluteError()\n",
104+
" ])\n",
105+
" return model\n",
102106
"\n",
103107
"model = create_model(passengers)"
104108
]

0 commit comments

Comments
 (0)