arviz_stats.base.array_stats.kde

Contents

arviz_stats.base.array_stats.kde#

array_stats.kde(ary, axis=-1, circular=False, grid_len=512, **kwargs)#

Compute KDE on array-like inputs.

Parameters:
aryarray_like
axisint, sequence of int or None, default -1
circularbool, default False
grid_lenint, default 512
**kwargsdict, optional

Additional keyword arguments passed to the KDE implementation. Supported arguments include:

  • bwstr or float, optional

    The bandwidth of the kernel. Options include “scott” (default), “silverman”, “isj”, and “experimental”, or a positive float.

  • adaptivebool, optional

    Whether to use an adaptive KDE. Defaults to False.

Returns:
grid, pdf, bwarray_like

grid and pdf will have the same shape: the same as ary minus the dimensions in axis plus an extra dimension of length grid_len. Same for bw except it will not have the extra dimension.