From d8890d81dbdc3178343de78a3c65fe59129109f0 Mon Sep 17 00:00:00 2001 From: Drew Herren Date: Wed, 12 Feb 2025 20:00:22 -0600 Subject: [PATCH] Fixed tree inspection notebook --- demo/notebooks/tree_inspection.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/notebooks/tree_inspection.ipynb b/demo/notebooks/tree_inspection.ipynb index 0c9149c4..089baf49 100644 --- a/demo/notebooks/tree_inspection.ipynb +++ b/demo/notebooks/tree_inspection.ipynb @@ -134,7 +134,7 @@ "metadata": {}, "outputs": [], "source": [ - "forest_preds_y_mcmc = bart_model.y_hat_test[:,bart_model.num_gfr:]\n", + "forest_preds_y_mcmc = bart_model.y_hat_test\n", "y_avg_mcmc = np.squeeze(forest_preds_y_mcmc).mean(axis = 1, keepdims = True)\n", "y_df_mcmc = pd.DataFrame(np.concatenate((np.expand_dims(y_test,1), y_avg_mcmc), axis = 1), columns=[\"True outcome\", \"Average estimated outcome\"])\n", "sns.scatterplot(data=y_df_mcmc, x=\"Average estimated outcome\", y=\"True outcome\")\n",