arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-01-14 12:14:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add missing custom_objects in load_model()

Browse Source
This commit is contained in:
Aurélien Geron
2024-10-05 18:53:13 +13:00
parent 2c0de8de67
commit 9143aa2fc4
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
12_custom_models_and_training_with_tensorflow.ipynb
View File

@@ -3365,7 +3365,10 @@
"source": [
"# extra code – the model can be loaded and you can continue training or use it\n",
"# to make predictions\n",
"model = tf.keras.models.load_model(\"my_custom_model.keras\")\n",
"model = tf.keras.models.load_model(\n",
" \"my_custom_model.keras\",\n",
" custom_objects={\"ResidualRegressor\": ResidualRegressor}\n",
")\n",
"history = model.fit(X_train_scaled, y_train, epochs=2)\n",
"model.predict(X_test_scaled[:3])"
]