Prefer operating on ValuationResult over dataframe for illustrative purposes

Author: mdbenito

notebooks/msr_banzhaf_digits.ipynb

@@ -30,7 +30,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {
     "tags": [
      "hide"
@@ -52,6 +52,7 @@
     "import numpy as np\n",
     "import torch\n",
     "from support.banzhaf import load_digits_dataset\n",
+    "from support.common import filecache\n",
     "\n",
     "from pydvl.reporting.plots import plot_result_errors\n",
     "\n",
@@ -152,7 +153,8 @@
     "    test = test[:10]\n",
     "    n_jobs = 1\n",
     "    n_epochs = 1\n",
-    "    batch_size = 1"
+    "    batch_size = 1\n",
+    "    filecache = lambda x: lambda y: y  # passthrough"
    ]
   },
   {
@@ -279,7 +281,6 @@
    "source": [
     "from joblib import parallel_config\n",
     "\n",
-    "from notebooks.support.common import filecache\n",
     "from pydvl.valuation.methods import BanzhafValuation\n",
     "\n",
     "valuation = BanzhafValuation(utility, sampler=sampler, is_done=stopping, progress=True)\n",