data_juicer.analysis.draw module

data_juicer.analysis.draw.draw_heatmap(data, xlabels, ylabels='auto', figsize=None, triangle=False, show=False)

Draw heatmap of input data with special labels.

Parameters:

• data – input data, now support [list, tuple, numpy array, ‘torch tensor’]

• xlabels – x axis labels.

• ylabels – y axis labels, if None, use xlabels.

• figsize – figure size.

• triangle – only display triangle.

Returns:

a plot figure.