Merge pull request #1149 from facebookexperimental/viz_accuracy

Fix feature plotter instance creation
facebookexperimental · Nov 18, 2024 · 96c0358 · 96c0358
2 parents 648d48d + 86d1c98
commit 96c0358
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/src/robyn/tutorials/tutorial1_src.ipynb b/python/src/robyn/tutorials/tutorial1_src.ipynb
@@ -215,14 +215,14 @@
     "%matplotlib inline\n",
     "\n",
     "# Create a FeaturePlotter instance\n",
-    "feature_plotter = FeaturePlotter(mmm_data, hyperparameters)\n",
+    "feature_plotter = FeaturePlotter(mmm_data, hyperparameters, featurized_mmm_data)\n",
     "# Extract the list of results\n",
     "results_list = featurized_mmm_data.modNLS[\"results\"]\n",
     "# Plot spend-exposure relationship for each channel in the results\n",
     "for result in results_list:\n",
     "    channel = result[\"channel\"]\n",
     "    try:\n",
-    "        fig = feature_plotter.plot_spend_exposure(featurized_mmm_data, channel)\n",
+    "        fig = feature_plotter.plot_spend_exposure(channel)\n",
     "        plt.show()\n",
     "    except ValueError as e:\n",
     "        print(f\"Skipping {channel}: {str(e)}\")"