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.

参数:

• 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.

返回:

a plot figure.