xiaohei/model_garden
1
0
Fork 0
mirror of https://github.com/GoogleCloudPlatform/vertex-ai-samples.git synced 2026-09-26 14:42:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: exception (#1714)

Browse Source 
* fix: exception

* fix: exception

* fix: exception

* fix: exception
This commit is contained in:
Andrew Ferlitsch
2023-04-11 20:40:22 +00:00
committed by GitHub
parent 973ecf95e7
commit 1d6f9bc36b
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/official/ml_metadata/sdk-metric-parameter-tracking-for-locally-trained-models.ipynb
+10 -4
View File
@@ -604,12 +604,18 @@
" epochs=params[\"epochs\"],\n", " epochs=params[\"epochs\"],\n",
" dropout_rate=params[\"dropout_rate\"],\n", " dropout_rate=params[\"dropout_rate\"],\n",
" )\n", " )\n",
" aiplatform.log_metrics(\n", "\n",
" {metric: values[-1] for metric, values in history.history.items()}\n", " for metric, values in history.history.items():\n",
" )\n", " try:\n",
" aiplatform.log_metrics({metric: values[-1]})\n",
" except:\n",
" aiplatform.log_metrics({metric: 0.0})\n",
"\n", "\n",
" loss, mae, mse = model.evaluate(normed_test_data, test_labels, verbose=2)\n", " loss, mae, mse = model.evaluate(normed_test_data, test_labels, verbose=2)\n",
" aiplatform.log_metrics({\"eval_loss\": loss, \"eval_mae\": mae, \"eval_mse\": mse})" " try:\n",
" aiplatform.log_metrics({\"eval_loss\": loss, \"eval_mae\": mae, \"eval_mse\": mse})\n",
" except:\n",
" aiplatform.log_metrics({\"eval_loss\": 0.0, \"eval_mae\": 0.0, \"eval_mse\": 0.0})"
] ]
}, },
{ {