Merge pull request #2046 from recommenders-team/jun/fix_benchmark_las…

…t_cell Fix benchmarks last cell to store value, not [value]
recommenders-team · Dec 22, 2023 · 5bf18a0 · 5bf18a0
2 parents 82ee6d3 + 495d09d
commit 5bf18a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/06_benchmarks/movielens.ipynb b/examples/06_benchmarks/movielens.ipynb
@@ -1207,7 +1207,7 @@
             "source": [
                 "# Record results for tests - ignore this cell\n",
                 "for algo in algorithms:\n",
-                "    store_metadata(algo, df_results.loc[df_results[\"Algo\"] == algo, \"nDCG@k\"].values)\n"
+                "    store_metadata(algo, df_results.loc[df_results[\"Algo\"] == algo, \"nDCG@k\"].values[0])\n"
             ]
         }
     ],