Merge branch 'master' into ilmat/fix-min-data-bin

microsoft · Nov 3, 2021 · 0b3080a · 0b3080a
2 parents af11b31 + 5733b85
commit 0b3080a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/notebooks/Interpretability - Explanation Dashboard.ipynb b/notebooks/Interpretability - Explanation Dashboard.ipynb
@@ -169,7 +169,7 @@
     "    model=model,\n",
     "    targetCol=\"probability\",\n",
     "    targetClasses=[1],\n",
-    "    backgroundData=training.orderBy(rand()).limit(100).cache(),\n",
+    "    backgroundData=broadcast(training.orderBy(rand()).limit(100).cache()),\n",
     ")\n",
     "\n",
     "shap_df = shap.transform(explain_instances)\n"

diff --git a/notebooks/Interpretability - Tabular SHAP explainer.ipynb b/notebooks/Interpretability - Tabular SHAP explainer.ipynb
@@ -165,7 +165,7 @@
     "    model=model,\n",
     "    targetCol=\"probability\",\n",
     "    targetClasses=[1],\n",
-    "    backgroundData=training.orderBy(rand()).limit(100).cache(),\n",
+    "    backgroundData=broadcast(training.orderBy(rand()).limit(100).cache()),\n",
     ")\n",
     "\n",
     "shap_df = shap.transform(explain_instances)\n"