black

Author: OriolAbril

examples/generalized_linear_models/GLM-robust-with-outlier-detection.ipynb

@@ -1949,38 +1949,18 @@
     "    )\n",
     "\n",
     "# plot hogg outlier posterior distribution\n",
-    "gd.ax.plot(\n",
-    "    x,\n",
-    "    subsample_helper(trc_hogg.posterior[\"outlier_mean\"], 400),\n",
-    "    color=\"C3\",\n",
-    "    linewidth=0.5,\n",
-    "    alpha=0.2,\n",
-    "    zorder=1,\n",
-    ")\n",
+    "outlier_mean = subsample_helper(trc_hogg.posterior[\"outlier_mean\"], 400)\n",
+    "gd.ax.plot(x, outlier_mean, color=\"C3\", linewidth=0.5, alpha=0.2, zorder=1)\n",
     "\n",
     "# plot the 3 model (inlier) posterior distributions\n",
     "y_mean = subsample_helper(trc_ols.posterior[\"y_mean\"], 200)\n",
     "gd.ax.plot(x, y_mean, color=\"C0\", linewidth=0.5, alpha=0.2, zorder=2)\n",
     "\n",
     "y_mean = subsample_helper(trc_studentt.posterior[\"y_mean\"], 200)\n",
-    "gd.ax.plot(\n",
-    "    x,\n",
-    "    y_mean,\n",
-    "    color=\"C1\",\n",
-    "    linewidth=0.5,\n",
-    "    alpha=0.2,\n",
-    "    zorder=3,\n",
-    ")\n",
+    "gd.ax.plot(x, y_mean, color=\"C1\", linewidth=0.5, alpha=0.2, zorder=3)\n",
     "\n",
     "y_mean = subsample_helper(trc_hogg.posterior[\"y_mean\"], 200)\n",
-    "gd.ax.plot(\n",
-    "    x,\n",
-    "    y_mean,\n",
-    "    color=\"C2\",\n",
-    "    linewidth=0.5,\n",
-    "    alpha=0.2,\n",
-    "    zorder=4,\n",
-    ")\n",
+    "gd.ax.plot(x, y_mean, color=\"C2\", linewidth=0.5, alpha=0.2, zorder=4)\n",
     "\n",
     "# add legend for regression lines plotted above\n",
     "line_legend = plt.legend(\n",