carat.display.embedding_plot

carat.display.embedding_plot(data, clusters=None, ax=None, **kwargs)

Display an 2D or 3D embedding of the rhythmic patterns data.

Parameters:

data : np.ndarray

Low-embedding data points

ax : matplotlib.axes.Axes or None

Axes to plot on instead of the default plt.gca().

clusters : np.ndarray

Array indicating cluster number for each point of the input data. If provided (not None) the clusters area displayed with colors.

kwargs : additional keyword arguments

Arguments passed through to matplotlib.pyplot.pcolormesh.

Returns:

axes

The axis handle for the figure.

See also

matplotlib.pyplot.pcolormesh