fix xai lab

Theoneforschools · Jun 11, 2023 · 65ce0f0 · 65ce0f0
1 parent 22eb88e
commit 65ce0f0
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 441 deletions.
diff --git a/notebooks/ml_fairness_explainability/explainable_ai/labs/xai_image_vertex.ipynb b/notebooks/ml_fairness_explainability/explainable_ai/labs/xai_image_vertex.ipynb
@@ -427,9 +427,7 @@
     "        verbose=2,  # 0=silent, 1=progress bar, 2=one line per epoch\n",
     "    )\n",
     "\n",
-    "    tf.saved_model.save(\n",
-    "        obj=model, export_dir=output_dir\n",
-    "    )  # with default serving function\n",
+    "    model.save(output_dir)\n",
     "\n",
     "    print(\"Exported trained model to {}\".format(output_dir))\n",
     "\n",
@@ -487,7 +485,7 @@
     "    version='0.1',\n",
     "    packages=find_packages(),\n",
     "    include_package_data=True,\n",
-    "    install_requires=['fire==0.4.0'],\n",
+    "    install_requires=['fire==0.4.0', 'tensorflow-hub==0.12.0'],\n",
     "    description='Flowers image classifier training application.'\n",
     ")"
    ]
@@ -981,14 +979,14 @@
   },
   "environment": {
    "kernel": "python3",
-   "name": "tf2-gpu.2-6.m90",
+   "name": "tf2-gpu.2-8.m108",
    "type": "gcloud",
-   "uri": "gcr.io/deeplearning-platform-release/tf2-gpu.2-6:m90"
+   "uri": "gcr.io/deeplearning-platform-release/tf2-gpu.2-8:m108"
   },
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel) (Local)",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
-   "name": "local-python3"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {

diff --git a/notebooks/ml_fairness_explainability/explainable_ai/solutions/xai_image_vertex.ipynb b/notebooks/ml_fairness_explainability/explainable_ai/solutions/xai_image_vertex.ipynb