arviz 0.23.1__py3-none-any.whl → 1.0.0rc0__py3-none-any.whl

arviz/plots/backends/bokeh/essplot.py

@@ -1,176 +0,0 @@
-# pylint: disable=all
-"""Bokeh ESS plots."""
-import numpy as np
-from bokeh.models import ColumnDataSource, Span
-from bokeh.models.annotations import Legend, Title
-from scipy.stats import rankdata
-
-from .. import show_layout
-from . import backend_kwarg_defaults, create_axes_grid
-from ...plot_utils import _scale_fig_size
-from bokeh.models.glyphs import Scatter
-
-
-def plot_ess(
-    ax,
-    plotters,
-    xdata,
-    ess_tail_dataset,
-    mean_ess,
-    sd_ess,
-    idata,
-    data,
-    kind,
-    extra_methods,
-    textsize,
-    rows,
-    cols,
-    figsize,
-    kwargs,
-    extra_kwargs,
-    text_kwargs,
-    n_samples,
-    relative,
-    min_ess,
-    labeller,
-    ylabel,
-    rug,
-    rug_kind,
-    rug_kwargs,
-    hline_kwargs,
-    backend_kwargs,
-    show,
-):
-    """Bokeh essplot."""
-    if backend_kwargs is None:
-        backend_kwargs = {}
-
-    backend_kwargs = {
-        **backend_kwarg_defaults(),
-        **backend_kwargs,
-    }
-
-    (figsize, *_, _linewidth, _markersize) = _scale_fig_size(figsize, textsize, rows, cols)
-
-    if hline_kwargs is None:
-        hline_kwargs = {}
-    hline_kwargs.setdefault("line_color", "red")
-    hline_kwargs.setdefault("line_width", 3)
-    hline_kwargs.setdefault("line_dash", "dashed")
-    hline_kwargs.setdefault("line_alpha", 1)
-
-    if ax is None:
-        ax = create_axes_grid(
-            len(plotters),
-            rows,
-            cols,
-            figsize=figsize,
-            backend_kwargs=backend_kwargs,
-        )
-    else:
-        ax = np.atleast_2d(ax)
-
-    for (var_name, selection, isel, x), ax_ in zip(
-        plotters, (item for item in ax.flatten() if item is not None)
-    ):
-        bulk_points = ax_.scatter(np.asarray(xdata), np.asarray(x), marker="circle", size=6)
-        if kind == "evolution":
-            bulk_line = ax_.line(np.asarray(xdata), np.asarray(x))
-            ess_tail = ess_tail_dataset[var_name].sel(**selection)
-            tail_points = ax_.line(np.asarray(xdata), np.asarray(ess_tail), color="orange")
-            tail_line = ax_.scatter(
-                np.asarray(xdata), np.asarray(ess_tail), marker="circle", size=6, color="orange"
-            )
-        elif rug:
-            if rug_kwargs is None:
-                rug_kwargs = {}
-            if not hasattr(idata, "sample_stats"):
-                raise ValueError("InferenceData object must contain sample_stats for rug plot")
-            if not hasattr(idata.sample_stats, rug_kind):
-                raise ValueError(f"InferenceData does not contain {rug_kind} data")
-
-            rug_kwargs.setdefault("space", 0.1)
-            _rug_kwargs = {}
-            _rug_kwargs.setdefault("size", 8)
-            _rug_kwargs.setdefault("line_color", rug_kwargs.get("line_color", "black"))
-            _rug_kwargs.setdefault("line_width", 1)
-            _rug_kwargs.setdefault("line_alpha", 0.35)
-            _rug_kwargs.setdefault("angle", np.pi / 2)
-
-            values = data[var_name].sel(**selection).values.flatten()
-            mask = idata.sample_stats[rug_kind].values.flatten()
-            values = rankdata(values, method="average")[mask]
-            rug_space = np.max(x) * rug_kwargs.pop("space")
-            rug_x, rug_y = values / (len(mask) - 1), np.zeros_like(values) - rug_space
-
-            glyph = Scatter(x="rug_x", y="rug_y", marker="dash", **_rug_kwargs)
-            cds_rug = ColumnDataSource({"rug_x": np.asarray(rug_x), "rug_y": np.asarray(rug_y)})
-            ax_.add_glyph(cds_rug, glyph)
-
-            hline = Span(
-                location=0,
-                dimension="width",
-                line_color="black",
-                line_width=_linewidth,
-                line_alpha=0.7,
-            )
-
-            ax_.renderers.append(hline)
-
-        if extra_methods:
-            mean_ess_i = mean_ess[var_name].sel(**selection).values.item()
-            sd_ess_i = sd_ess[var_name].sel(**selection).values.item()
-
-            hline = Span(
-                location=mean_ess_i,
-                dimension="width",
-                line_color="black",
-                line_width=2,
-                line_dash="dashed",
-                line_alpha=1.0,
-            )
-
-            ax_.renderers.append(hline)
-
-            hline = Span(
-                location=sd_ess_i,
-                dimension="width",
-                line_color="black",
-                line_width=1,
-                line_dash="dashed",
-                line_alpha=1.0,
-            )
-
-            ax_.renderers.append(hline)
-
-        if relative and kind == "evolution":
-            thin_xdata = np.linspace(xdata.min(), xdata.max(), 100)
-            ax_.line(thin_xdata, min_ess / thin_xdata, **hline_kwargs)
-        else:
-            hline = Span(
-                location=min_ess / n_samples if relative else min_ess,
-                dimension="width",
-                **hline_kwargs,
-            )
-
-            ax_.renderers.append(hline)
-
-        if kind == "evolution":
-            legend = Legend(
-                items=[("bulk", [bulk_points, bulk_line]), ("tail", [tail_line, tail_points])],
-                location="center_right",
-                orientation="horizontal",
-            )
-            ax_.add_layout(legend, "above")
-            ax_.legend.click_policy = "hide"
-
-        title = Title()
-        title.text = labeller.make_label_vert(var_name, selection, isel)
-        ax_.title = title
-
-        ax_.xaxis.axis_label = "Total number of draws" if kind == "evolution" else "Quantile"
-        ax_.yaxis.axis_label = ylabel.format("Relative ESS" if relative else "ESS")
-
-    show_layout(ax, show)
-
-    return ax