idmd.visualization.visualizer.DataVisualizer

class idmd.visualization.visualizer.DataVisualizer[source]

Bases: object

Utility class for generating visualizations.

Methods

generate_bar_plot

Generates a bar plot using PlotGenerator.

generate_correlation_heatmap

Generates a correlation heatmap using HeatmapGenerator.

generate_histograms

Generates histograms using HistogramGenerator.

generate_line_plot

Generates a line plot using PlotGenerator.

generate_overview

Generates an overview of the dataset, including data types and plottability.

static generate_bar_plot(df, column)[source]

Generates a bar plot using PlotGenerator.

Args:

df (pd.DataFrame): The DataFrame containing the data. column (str): The column to include in the bar plot.

Returns:

plt.Figure: The generated bar plot.

static generate_correlation_heatmap(df, columns)[source]

Generates a correlation heatmap using HeatmapGenerator.

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.

static generate_histograms(df, columns)[source]

Generates histograms using HistogramGenerator.

Args:

df (pd.DataFrame): The DataFrame containing the data. columns (List[str]): The columns to include in the histograms.

Returns:

plt.Figure: The generated histograms.

static generate_line_plot(df, columns)[source]

Generates a line plot using PlotGenerator.

Args:

df (pd.DataFrame): The DataFrame containing the data. columns (List[str]): The columns to include in the line plot.

Returns:

plt.Figure: The generated line plot.

static generate_overview(df)[source]

Generates an overview of the dataset, including data types and plottability.

Return type:

DataFrame

Args:

df (pd.DataFrame): The DataFrame to analyze.

Returns:

pd.DataFrame: A DataFrame containing the overview information.