idmd.visualization.heatmaps.HeatmapGenerator

class idmd.visualization.heatmaps.HeatmapGenerator[source]

Bases: object

Generates correlation heatmaps.

Methods

generate_correlation_heatmap

Generates a correlation heatmap for the specified columns.

static generate_correlation_heatmap(df, columns)[source]

Generates a correlation heatmap for the specified columns.

Return type:: Figure

Args:: df (pd.DataFrame): The DataFrame containing the data. columns (List[str]): The columns to include in the heatmap.
Returns:: plt.Figure: The generated heatmap.