Spaces:
Sleeping
Sleeping
Delete src
Browse files- src/AutoVisualizer/__init__.py +0 -0
- src/AutoVisualizer/categorical_viz.py +0 -283
- src/AutoVisualizer/numerical_viz.py +0 -233
- src/AutoVisualizer/processing.py +0 -250
- src/app.py +0 -221
src/AutoVisualizer/__init__.py
DELETED
|
File without changes
|
src/AutoVisualizer/categorical_viz.py
DELETED
|
@@ -1,283 +0,0 @@
|
|
| 1 |
-
# Module that will handle plots like bar chart, count plot, pie chart, etc.
|
| 2 |
-
import plotly.express as px
|
| 3 |
-
import streamlit as st
|
| 4 |
-
import numpy as np
|
| 5 |
-
import pandas as pd
|
| 6 |
-
from plotly.subplots import make_subplots
|
| 7 |
-
import plotly.graph_objs as go
|
| 8 |
-
|
| 9 |
-
# Subplotting Function
|
| 10 |
-
def combine_figures_as_subplots(figures: list, rows_per_column: int = 2):
|
| 11 |
-
"""
|
| 12 |
-
Combines a list of Plotly figures into a single subplot layout.
|
| 13 |
-
|
| 14 |
-
Args:
|
| 15 |
-
figures (list): A list of individual Plotly figure objects to be arranged as subplots.
|
| 16 |
-
rows_per_column (int, optional): The number of subplot rows per column. Defaults to 2.
|
| 17 |
-
|
| 18 |
-
Returns:
|
| 19 |
-
plotly.graph_objects.Figure: A single Plotly figure containing all input figures as subplots.
|
| 20 |
-
"""
|
| 21 |
-
total = len(figures)
|
| 22 |
-
cols = 1 if total <= rows_per_column else 2
|
| 23 |
-
rows = (total + cols - 1) // cols # Ceiling division to compute total rows
|
| 24 |
-
|
| 25 |
-
# Determine subplot types for each cell
|
| 26 |
-
specs = []
|
| 27 |
-
for i in range(rows):
|
| 28 |
-
row_specs = []
|
| 29 |
-
for j in range(cols):
|
| 30 |
-
idx = i * cols + j
|
| 31 |
-
if idx < total and figures[idx].data and figures[idx].data[0].type == "pie":
|
| 32 |
-
row_specs.append({"type": "domain"})
|
| 33 |
-
else:
|
| 34 |
-
row_specs.append({"type": "xy"})
|
| 35 |
-
specs.append(row_specs)
|
| 36 |
-
|
| 37 |
-
# Create subplots with correct specs
|
| 38 |
-
subplot_fig = make_subplots(
|
| 39 |
-
rows=rows, cols=cols,
|
| 40 |
-
subplot_titles=[fig.layout.title.text for fig in figures],
|
| 41 |
-
specs=specs
|
| 42 |
-
)
|
| 43 |
-
|
| 44 |
-
# Add each figure's traces to the corresponding subplot cell
|
| 45 |
-
for idx, fig in enumerate(figures):
|
| 46 |
-
row = (idx // cols) + 1
|
| 47 |
-
col = (idx % cols) + 1
|
| 48 |
-
|
| 49 |
-
for trace in fig.data:
|
| 50 |
-
subplot_fig.add_trace(trace, row=row, col=col)
|
| 51 |
-
|
| 52 |
-
# Only update axes for non-pie charts
|
| 53 |
-
if fig.data[0].type != "pie":
|
| 54 |
-
xaxis_title = fig.layout.xaxis.title.text
|
| 55 |
-
yaxis_title = fig.layout.yaxis.title.text
|
| 56 |
-
subplot_fig.update_xaxes(title_text=xaxis_title, row=row, col=col)
|
| 57 |
-
subplot_fig.update_yaxes(title_text=yaxis_title, row=row, col=col)
|
| 58 |
-
|
| 59 |
-
# Final layout adjustments
|
| 60 |
-
subplot_fig.update_layout(height=350 * rows, showlegend=False)
|
| 61 |
-
return subplot_fig
|
| 62 |
-
|
| 63 |
-
# Count Plots Function
|
| 64 |
-
def generate_count_plots(df: pd.DataFrame, categorical_column: str, max_label_len: int = 10):
|
| 65 |
-
"""
|
| 66 |
-
Generates a count plot (bar chart) for a given categorical column in a DataFrame.
|
| 67 |
-
|
| 68 |
-
Args:
|
| 69 |
-
df (pd.DataFrame): The input dataset.
|
| 70 |
-
categorical_column (str): The name of the categorical column to plot.
|
| 71 |
-
max_label_len (int, optional): Maximum number of characters to display on x-axis labels.
|
| 72 |
-
Longer labels are truncated with an ellipsis. Defaults to 10.
|
| 73 |
-
|
| 74 |
-
Returns:
|
| 75 |
-
list: A list containing a single Plotly bar chart figure.
|
| 76 |
-
"""
|
| 77 |
-
plots = []
|
| 78 |
-
# Compute value counts for the selected column
|
| 79 |
-
value_counts = df[categorical_column].value_counts().reset_index()
|
| 80 |
-
value_counts.columns = [categorical_column, "Count"]
|
| 81 |
-
|
| 82 |
-
# Add truncated label column
|
| 83 |
-
value_counts["Truncated"] = value_counts[categorical_column].apply(
|
| 84 |
-
lambda x: x if len(str(x)) <= max_label_len else str(x)[:max_label_len] + "…"
|
| 85 |
-
)
|
| 86 |
-
|
| 87 |
-
# Create a bar plot
|
| 88 |
-
fig = px.bar(
|
| 89 |
-
value_counts, x="Truncated", y="Count", color="Truncated",
|
| 90 |
-
title=f"Count Plot for '{categorical_column}'",
|
| 91 |
-
labels={categorical_column: "Truncated", "Count": "Frequency"},
|
| 92 |
-
custom_data=[value_counts[categorical_column]]
|
| 93 |
-
)
|
| 94 |
-
|
| 95 |
-
# Use full label in hover template
|
| 96 |
-
fig.update_traces(
|
| 97 |
-
hovertemplate=f"{categorical_column}=%{{customdata}}<br>Count=%{{y}}<extra></extra>"
|
| 98 |
-
)
|
| 99 |
-
|
| 100 |
-
fig.update_layout(showlegend=False, xaxis_title=categorical_column)
|
| 101 |
-
plots.append(fig)
|
| 102 |
-
return plots
|
| 103 |
-
|
| 104 |
-
# Bar Plot Function
|
| 105 |
-
def generate_bar_plots(df: pd.DataFrame, x_col: str, y_columns: list, max_label_len: int = 10):
|
| 106 |
-
"""
|
| 107 |
-
Generates bar plots showing the average of continuous features grouped by a categorical column.
|
| 108 |
-
|
| 109 |
-
Args:
|
| 110 |
-
df (pd.DataFrame): The input dataset.
|
| 111 |
-
x_col (str): The categorical column to group by (x-axis).
|
| 112 |
-
y_columns (list): List of continuous columns to aggregate and plot (y-axis).
|
| 113 |
-
max_label_len (int, optional): Maximum number of characters for x-axis labels.
|
| 114 |
-
Labels longer than this are truncated. Defaults to 10.
|
| 115 |
-
|
| 116 |
-
Returns:
|
| 117 |
-
list: A list of Plotly bar chart figures.
|
| 118 |
-
"""
|
| 119 |
-
plots = []
|
| 120 |
-
|
| 121 |
-
for y_col in y_columns:
|
| 122 |
-
# Compute mean aggregation
|
| 123 |
-
agg_func = df.groupby(x_col)[y_col].mean().reset_index()
|
| 124 |
-
|
| 125 |
-
# Add truncated label for x-axis ticks
|
| 126 |
-
agg_func["Truncated"] = agg_func[x_col].apply(
|
| 127 |
-
lambda x: x if len(str(x)) <= max_label_len else str(x)[:max_label_len] + "…"
|
| 128 |
-
)
|
| 129 |
-
|
| 130 |
-
fig = px.bar(
|
| 131 |
-
agg_func, x="Truncated", y=y_col, color="Truncated", # color by original to retain legend info
|
| 132 |
-
title=f"{x_col} vs {y_col} (Average)",
|
| 133 |
-
labels={"Truncated": "", y_col: y_col},
|
| 134 |
-
custom_data=[agg_func[x_col]],
|
| 135 |
-
)
|
| 136 |
-
|
| 137 |
-
# Format hover values
|
| 138 |
-
if (df[y_col].dropna() % 1 == 0).all():
|
| 139 |
-
hover_format = ".0f"
|
| 140 |
-
else:
|
| 141 |
-
hover_format = ".2f"
|
| 142 |
-
|
| 143 |
-
# Show full x value on hover instead of truncated
|
| 144 |
-
fig.update_traces(
|
| 145 |
-
hovertemplate=f"{x_col}=%{{customdata[0]}}<br>Average {y_col}=%{{y:{hover_format}}}<extra></extra>",
|
| 146 |
-
)
|
| 147 |
-
|
| 148 |
-
fig.update_layout(showlegend=False, xaxis_title=x_col)
|
| 149 |
-
plots.append(fig)
|
| 150 |
-
|
| 151 |
-
return plots
|
| 152 |
-
|
| 153 |
-
# Grouped Bar Plot Function
|
| 154 |
-
def generate_grouped_bar_plots(df: pd.DataFrame, x_columns: list, y_columns: list, max_label_len: int = 10):
|
| 155 |
-
"""
|
| 156 |
-
Generates grouped bar plots showing the average of continuous features
|
| 157 |
-
grouped by a categorical feature and further separated by a binary hue column.
|
| 158 |
-
|
| 159 |
-
Args:
|
| 160 |
-
df (pd.DataFrame): The input dataset.
|
| 161 |
-
x_columns (list): Categorical columns to consider for x-axis and hue roles.
|
| 162 |
-
y_columns (list): Continuous columns to be averaged and plotted.
|
| 163 |
-
max_label_len (int, optional): Maximum length of x-axis labels before truncation. Defaults to 10.
|
| 164 |
-
|
| 165 |
-
Returns:
|
| 166 |
-
list: A list of Plotly grouped bar chart figures.
|
| 167 |
-
"""
|
| 168 |
-
plots = []
|
| 169 |
-
|
| 170 |
-
# Split categorical columns
|
| 171 |
-
hue_candidates = [col for col in x_columns if df[col].nunique() <= 2]
|
| 172 |
-
x_candidates = [col for col in x_columns if df[col].nunique() > 2 and df[col].nunique() <= 10]
|
| 173 |
-
|
| 174 |
-
for x_col in x_candidates:
|
| 175 |
-
for y_col in y_columns:
|
| 176 |
-
for hue_col in hue_candidates:
|
| 177 |
-
# Group and aggregate
|
| 178 |
-
agg_func = df.groupby([x_col, hue_col])[y_col].mean().reset_index()
|
| 179 |
-
|
| 180 |
-
# Add truncated label for x-axis ticks
|
| 181 |
-
agg_func["Truncated"] = agg_func[x_col].apply(
|
| 182 |
-
lambda x: x if len(str(x)) <= max_label_len else str(x)[:max_label_len] + "…"
|
| 183 |
-
)
|
| 184 |
-
|
| 185 |
-
fig = px.bar(
|
| 186 |
-
agg_func,
|
| 187 |
-
x="Truncated",
|
| 188 |
-
y=y_col,
|
| 189 |
-
color=hue_col,
|
| 190 |
-
barmode="group",
|
| 191 |
-
title=f"{x_col} vs {y_col} grouped by {hue_col}",
|
| 192 |
-
labels={x_col: x_col, y_col: f"Average {y_col}", hue_col: hue_col},
|
| 193 |
-
custom_data=[agg_func[x_col], agg_func[hue_col]]
|
| 194 |
-
)
|
| 195 |
-
|
| 196 |
-
# Formatting hover text
|
| 197 |
-
if (df[y_col].dropna() % 1 == 0).all():
|
| 198 |
-
hover_format = ".0f"
|
| 199 |
-
else:
|
| 200 |
-
hover_format = ".2f"
|
| 201 |
-
|
| 202 |
-
# Show full x value on hover instead of truncated
|
| 203 |
-
fig.update_traces(
|
| 204 |
-
hovertemplate=f"{x_col}=%{{customdata[0]}}<br>{hue_col}=%{{customdata[1]}}<br>Average {y_col}=%{{y:{hover_format}}}<extra></extra>",
|
| 205 |
-
)
|
| 206 |
-
fig.update_layout(showlegend=False, xaxis_title=x_col)
|
| 207 |
-
plots.append(fig)
|
| 208 |
-
|
| 209 |
-
return plots
|
| 210 |
-
|
| 211 |
-
# Pie Charts Function
|
| 212 |
-
def generate_pie_plots(df: pd.DataFrame, categorical_column: str, max_label_len: int = 10):
|
| 213 |
-
"""
|
| 214 |
-
Generates a pie chart showing the distribution of values for a given categorical column.
|
| 215 |
-
|
| 216 |
-
Args:
|
| 217 |
-
df (pd.DataFrame): The input dataset.
|
| 218 |
-
categorical_column (str): The categorical column for which to create a pie chart.
|
| 219 |
-
max_label_len (int, optional): Maximum length of labels (currently unused here but reserved for consistency). Defaults to 10.
|
| 220 |
-
|
| 221 |
-
Returns:
|
| 222 |
-
list: A list containing one Plotly pie chart figure.
|
| 223 |
-
"""
|
| 224 |
-
plots = []
|
| 225 |
-
# Count occurrences of each category
|
| 226 |
-
value_counts = df[categorical_column].value_counts().reset_index()
|
| 227 |
-
value_counts.columns = [categorical_column, "Count"]
|
| 228 |
-
|
| 229 |
-
fig = px.pie(
|
| 230 |
-
value_counts, names=categorical_column, values="Count",
|
| 231 |
-
hole=0.3,
|
| 232 |
-
title=f"Pie Plot for '{categorical_column}'",
|
| 233 |
-
)
|
| 234 |
-
# Use full label in hover template
|
| 235 |
-
fig.update_traces(
|
| 236 |
-
textinfo='value',
|
| 237 |
-
hovertemplate=f"{categorical_column}=%{{label}}<br>Percentage=%{{percent}}<extra></extra>"
|
| 238 |
-
)
|
| 239 |
-
|
| 240 |
-
fig.update_layout(showlegend=False)
|
| 241 |
-
plots.append(fig)
|
| 242 |
-
return plots
|
| 243 |
-
|
| 244 |
-
# HeatMap Function
|
| 245 |
-
# Heatmap showing correlated categorical features based on a numerical target column
|
| 246 |
-
def generate_categorical_correlation_heatmap(df: pd.DataFrame, numerical_col: str, categorical_columns: list):
|
| 247 |
-
"""
|
| 248 |
-
Generates a heatmap showing correlations between categorical features based on a numeric target column.
|
| 249 |
-
|
| 250 |
-
This function encodes each categorical feature by replacing its categories with the mean value
|
| 251 |
-
of the numeric column for that category. Then it computes the Pearson correlation between the
|
| 252 |
-
encoded categorical features.
|
| 253 |
-
|
| 254 |
-
Args:
|
| 255 |
-
df (pd.DataFrame): The input DataFrame.
|
| 256 |
-
numerical_col (str): The target numeric column to base encoding on.
|
| 257 |
-
categorical_columns (list): List of categorical feature names to analyze.
|
| 258 |
-
|
| 259 |
-
Returns:
|
| 260 |
-
list: A list containing one Plotly heatmap figure visualizing the correlation matrix.
|
| 261 |
-
"""
|
| 262 |
-
plots = []
|
| 263 |
-
encoded_df = pd.DataFrame()
|
| 264 |
-
|
| 265 |
-
# Encode categorical features using mean of numeric target per category
|
| 266 |
-
for col in categorical_columns:
|
| 267 |
-
temp = df[[col, numerical_col]].dropna()
|
| 268 |
-
means = temp.groupby(col)[numerical_col].mean()
|
| 269 |
-
encoded_df[col] = temp[col].map(means)
|
| 270 |
-
|
| 271 |
-
# Compute correlation matrix on the encoded values
|
| 272 |
-
corr_matrix = encoded_df.corr(method="pearson")
|
| 273 |
-
|
| 274 |
-
fig = px.imshow(
|
| 275 |
-
corr_matrix,
|
| 276 |
-
text_auto=True,
|
| 277 |
-
color_continuous_scale="RdBu",
|
| 278 |
-
aspect="auto",
|
| 279 |
-
title=f"Categorical Feature Correlation Heatmap Based on your Target : '{numerical_col}'",
|
| 280 |
-
)
|
| 281 |
-
|
| 282 |
-
plots.append(fig)
|
| 283 |
-
return plots
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
src/AutoVisualizer/numerical_viz.py
DELETED
|
@@ -1,233 +0,0 @@
|
|
| 1 |
-
# Module that will handle plots like histogram, boxplot, density plot, etc.
|
| 2 |
-
import plotly.express as px
|
| 3 |
-
import streamlit as st
|
| 4 |
-
import numpy as np
|
| 5 |
-
import pandas as pd
|
| 6 |
-
import plotly.graph_objs as go
|
| 7 |
-
from scipy.stats import gaussian_kde
|
| 8 |
-
|
| 9 |
-
# Box Plot Function
|
| 10 |
-
def generate_box_plots(df: pd.DataFrame, x_col: str, y_columns: list, max_label_len: int = 10):
|
| 11 |
-
"""
|
| 12 |
-
Generates box plots for multiple numerical columns grouped by a categorical column.
|
| 13 |
-
|
| 14 |
-
This function creates a series of Plotly box plots to visualize the distribution,
|
| 15 |
-
spread, and potential outliers of each numerical column across the values of a given
|
| 16 |
-
categorical column. Long category labels are truncated for better readability.
|
| 17 |
-
|
| 18 |
-
Args:
|
| 19 |
-
df (pd.DataFrame): The input DataFrame.
|
| 20 |
-
x_col (str): Categorical column to group the box plots by.
|
| 21 |
-
y_columns (list): List of numerical columns to plot on the y-axis.
|
| 22 |
-
max_label_len (int, optional): Maximum character length for category labels on the x-axis (default is 10).
|
| 23 |
-
|
| 24 |
-
Returns:
|
| 25 |
-
list: A list of Plotly figure objects, each representing a box plot for a y_column grouped by x_col.
|
| 26 |
-
"""
|
| 27 |
-
plots = []
|
| 28 |
-
|
| 29 |
-
# Drop rows with missing values in relevant columns
|
| 30 |
-
df = df[[x_col] + y_columns].dropna()
|
| 31 |
-
|
| 32 |
-
# Truncate long category names
|
| 33 |
-
df["Truncated_X"] = df[x_col].apply(
|
| 34 |
-
lambda x: str(x) if len(str(x)) <= max_label_len else str(x)[:max_label_len] + "…"
|
| 35 |
-
)
|
| 36 |
-
|
| 37 |
-
for y_col in y_columns:
|
| 38 |
-
fig = px.box(
|
| 39 |
-
df,
|
| 40 |
-
x="Truncated_X", y=y_col,
|
| 41 |
-
color="Truncated_X",
|
| 42 |
-
points="outliers", # Show outliers
|
| 43 |
-
title=f"Box Plot of {y_col} by {x_col}",
|
| 44 |
-
labels={"Truncated_X": x_col, y_col: y_col},
|
| 45 |
-
)
|
| 46 |
-
|
| 47 |
-
fig.update_layout(showlegend=False)
|
| 48 |
-
plots.append(fig)
|
| 49 |
-
|
| 50 |
-
return plots
|
| 51 |
-
|
| 52 |
-
# HeatMap Functions
|
| 53 |
-
## Heatmap of numerical (continuous) features
|
| 54 |
-
def generate_numeric_correlation_heatmap(df: pd.DataFrame):
|
| 55 |
-
"""
|
| 56 |
-
Generates a heatmap to visualize Pearson correlation between numerical features.
|
| 57 |
-
|
| 58 |
-
This function computes the correlation matrix for all numeric columns in the
|
| 59 |
-
DataFrame and returns a heatmap figure representing the strength and direction
|
| 60 |
-
of linear relationships between features.
|
| 61 |
-
|
| 62 |
-
Args:
|
| 63 |
-
df (pd.DataFrame): The input DataFrame.
|
| 64 |
-
|
| 65 |
-
Returns:
|
| 66 |
-
list: A list containing a single Plotly heatmap figure if enough numeric
|
| 67 |
-
features exist; otherwise, returns an empty list.
|
| 68 |
-
"""
|
| 69 |
-
plots = []
|
| 70 |
-
|
| 71 |
-
# Filter numeric columns
|
| 72 |
-
numeric_df = df.select_dtypes(include=np.number)
|
| 73 |
-
|
| 74 |
-
if numeric_df.shape[1] < 2:
|
| 75 |
-
return plots # Not enough numeric features to compute correlation
|
| 76 |
-
|
| 77 |
-
corr_matrix = numeric_df.corr(method="pearson")
|
| 78 |
-
|
| 79 |
-
fig = px.imshow(
|
| 80 |
-
corr_matrix,
|
| 81 |
-
text_auto=True,
|
| 82 |
-
color_continuous_scale="RdBu",
|
| 83 |
-
aspect="auto",
|
| 84 |
-
title="Numerical Feature Correlation Heatmap",
|
| 85 |
-
)
|
| 86 |
-
|
| 87 |
-
plots.append(fig)
|
| 88 |
-
return plots
|
| 89 |
-
|
| 90 |
-
# Scatter Plot Function
|
| 91 |
-
def generate_scatter_plots(df: pd.DataFrame, feature_pairs: list, color_by: str = None):
|
| 92 |
-
"""
|
| 93 |
-
Generates scatter plots for given pairs of numerical features.
|
| 94 |
-
|
| 95 |
-
Args:
|
| 96 |
-
df (pd.DataFrame): The input DataFrame containing the features.
|
| 97 |
-
feature_pairs (list): List of tuples containing (x, y) column names for each plot.
|
| 98 |
-
color_by (str, optional): Column name to color points by. Must be in df. Defaults to None.
|
| 99 |
-
|
| 100 |
-
Returns:
|
| 101 |
-
list: A list of Plotly scatter plot figures, one for each feature pair.
|
| 102 |
-
"""
|
| 103 |
-
plots = []
|
| 104 |
-
|
| 105 |
-
for x_col, y_col in feature_pairs:
|
| 106 |
-
fig = px.scatter(
|
| 107 |
-
df.dropna(subset=[x_col, y_col]),
|
| 108 |
-
x=x_col,
|
| 109 |
-
y=y_col,
|
| 110 |
-
color=color_by if color_by in df.columns else None,
|
| 111 |
-
title=f"Scatter Plot: {x_col} vs {y_col}",
|
| 112 |
-
labels={x_col: x_col, y_col: y_col}
|
| 113 |
-
)
|
| 114 |
-
# Update title and legend if coloring is applied
|
| 115 |
-
if color_by:
|
| 116 |
-
fig.update_layout(title = f"Scatter Plot: {x_col} vs {y_col} color by {color_by}",showlegend=bool(color_by))
|
| 117 |
-
|
| 118 |
-
fig.update_layout(showlegend=bool(color_by))
|
| 119 |
-
plots.append(fig)
|
| 120 |
-
|
| 121 |
-
return plots
|
| 122 |
-
|
| 123 |
-
# Histogram Function
|
| 124 |
-
def generate_histograms(df: pd.DataFrame, numeric_columns: list, bins: int = 30):
|
| 125 |
-
"""
|
| 126 |
-
Generates histogram and KDE plots for each specified numeric column.
|
| 127 |
-
|
| 128 |
-
Args:
|
| 129 |
-
df (pd.DataFrame): The input DataFrame containing numeric features.
|
| 130 |
-
numeric_columns (list): List of column names to plot histograms for.
|
| 131 |
-
bins (int, optional): Number of bins to use in the histogram. Defaults to 30.
|
| 132 |
-
|
| 133 |
-
Returns:
|
| 134 |
-
list: A list of Plotly figures, each showing a histogram and KDE curve.
|
| 135 |
-
"""
|
| 136 |
-
plots = []
|
| 137 |
-
|
| 138 |
-
for col in numeric_columns:
|
| 139 |
-
data = df[col].dropna()
|
| 140 |
-
|
| 141 |
-
# Histogram
|
| 142 |
-
hist = go.Histogram(
|
| 143 |
-
x=data,
|
| 144 |
-
nbinsx=bins,
|
| 145 |
-
name='Histogram',
|
| 146 |
-
opacity=0.6
|
| 147 |
-
)
|
| 148 |
-
|
| 149 |
-
# KDE Curve
|
| 150 |
-
kde = gaussian_kde(data)
|
| 151 |
-
x_range = np.linspace(data.min(), data.max(), 200)
|
| 152 |
-
kde_curve = go.Scatter(
|
| 153 |
-
x=x_range,
|
| 154 |
-
y=kde(x_range) * len(data) * (data.max() - data.min()) / bins, # scaled to match histogram
|
| 155 |
-
name='KDE',
|
| 156 |
-
mode='lines',
|
| 157 |
-
line=dict(color='red')
|
| 158 |
-
)
|
| 159 |
-
|
| 160 |
-
# Combine into a single figure
|
| 161 |
-
fig = go.Figure(data=[hist, kde_curve])
|
| 162 |
-
fig.update_layout(
|
| 163 |
-
title=f"Histogram + KDE for '{col}'",
|
| 164 |
-
xaxis_title=col,
|
| 165 |
-
yaxis_title="Count",
|
| 166 |
-
barmode='overlay',
|
| 167 |
-
showlegend=True
|
| 168 |
-
)
|
| 169 |
-
|
| 170 |
-
plots.append(fig)
|
| 171 |
-
|
| 172 |
-
return plots
|
| 173 |
-
|
| 174 |
-
# Line Plots Function
|
| 175 |
-
def generate_line_plots(df: pd.DataFrame, date_component_cols: list, y_columns: list, freq: str = "M"):
|
| 176 |
-
"""
|
| 177 |
-
Generates line plots for given numerical columns based on reconstructed datetime columns.
|
| 178 |
-
|
| 179 |
-
This function identifies datetime-related components in the DataFrame (e.g., 'order_year', 'order_month', 'order_day'),
|
| 180 |
-
reconstructs them into actual datetime objects, and then plots the specified `y_columns` over time using the chosen frequency.
|
| 181 |
-
|
| 182 |
-
Args:
|
| 183 |
-
df (pd.DataFrame): The input DataFrame containing date components and numerical data.
|
| 184 |
-
date_component_cols (list): List of column names representing datetime parts (e.g., 'order_date_year', 'order_date_month').
|
| 185 |
-
y_columns (list): List of numeric columns to be plotted against the date.
|
| 186 |
-
freq (str, optional): Resampling frequency for datetime aggregation.
|
| 187 |
-
Options:
|
| 188 |
-
'D' = daily
|
| 189 |
-
'W' = weekly
|
| 190 |
-
'M' = monthly (default)
|
| 191 |
-
'Y' = yearly
|
| 192 |
-
|
| 193 |
-
Returns:
|
| 194 |
-
list: A list of Plotly line plot figures for each (prefix, y_column) pair.
|
| 195 |
-
"""
|
| 196 |
-
from collections import defaultdict
|
| 197 |
-
plots = []
|
| 198 |
-
|
| 199 |
-
# Group columns by prefix
|
| 200 |
-
groups = defaultdict(dict)
|
| 201 |
-
for col in date_component_cols:
|
| 202 |
-
for suffix in ['_year', '_month', '_day', '_weekday']:
|
| 203 |
-
if col.endswith(suffix):
|
| 204 |
-
prefix = col.replace(suffix, '')
|
| 205 |
-
groups[prefix][suffix] = col
|
| 206 |
-
|
| 207 |
-
# Reconstruct datetime and generate plots
|
| 208 |
-
for prefix, components in groups.items():
|
| 209 |
-
if all(k in components for k in ['_year', '_month', '_day']):
|
| 210 |
-
# Build a datetime column from year, month, day
|
| 211 |
-
temp_df = df[[components['_year'], components['_month'], components['_day']] + y_columns].dropna()
|
| 212 |
-
temp_df["__date__"] = pd.to_datetime({
|
| 213 |
-
'year': temp_df[components['_year']],
|
| 214 |
-
'month': temp_df[components['_month']],
|
| 215 |
-
'day': temp_df[components['_day']]
|
| 216 |
-
}, errors='coerce')
|
| 217 |
-
temp_df = temp_df.dropna(subset=["__date__"])
|
| 218 |
-
temp_df.set_index("__date__", inplace=True)
|
| 219 |
-
|
| 220 |
-
# Resample and plot
|
| 221 |
-
resampled_df = temp_df.resample(freq).mean().reset_index()
|
| 222 |
-
|
| 223 |
-
for col in y_columns:
|
| 224 |
-
fig = px.line(
|
| 225 |
-
resampled_df,
|
| 226 |
-
x="__date__",
|
| 227 |
-
y=col,
|
| 228 |
-
title=f"Line Plot: {prefix} vs {col}",
|
| 229 |
-
labels={"__date__": "Date", col: col},
|
| 230 |
-
)
|
| 231 |
-
plots.append(fig)
|
| 232 |
-
|
| 233 |
-
return plots
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
src/AutoVisualizer/processing.py
DELETED
|
@@ -1,250 +0,0 @@
|
|
| 1 |
-
# Module that will handle processing tasks of the user dataset
|
| 2 |
-
import numpy as np
|
| 3 |
-
import pandas as pd
|
| 4 |
-
import streamlit as st
|
| 5 |
-
|
| 6 |
-
# A quick cleanliness checker function
|
| 7 |
-
def check_dataset_cleanliness(df):
|
| 8 |
-
issues_found = False
|
| 9 |
-
|
| 10 |
-
# 1. Columns with null values
|
| 11 |
-
null_cols = df.columns[df.isnull().any()].tolist()
|
| 12 |
-
if null_cols:
|
| 13 |
-
issues_found = True
|
| 14 |
-
st.warning(f"⚠️ These columns contain missing (NaN) values: {null_cols}")
|
| 15 |
-
|
| 16 |
-
# 2. Object columns that appear to be numeric (but aren't due to dirty values)
|
| 17 |
-
misclassified_numeric = []
|
| 18 |
-
for col in df.select_dtypes(include="object").columns:
|
| 19 |
-
non_null = df[col].dropna().astype(str)
|
| 20 |
-
sample_size = min(100, len(non_null))
|
| 21 |
-
if sample_size == 0:
|
| 22 |
-
continue # skip if column has no non-null values
|
| 23 |
-
sample = non_null.sample(sample_size, random_state=1)
|
| 24 |
-
numeric_like_ratio = sample.str.replace(",", "").str.replace(".", "", regex=False).str.isdigit().mean()
|
| 25 |
-
if numeric_like_ratio > 0.6:
|
| 26 |
-
misclassified_numeric.append(col)
|
| 27 |
-
|
| 28 |
-
if misclassified_numeric:
|
| 29 |
-
issues_found = True
|
| 30 |
-
st.warning(
|
| 31 |
-
f"⚠️ These columns are stored as `object` but mostly contain numeric values.\n"
|
| 32 |
-
f"This may be due to the presence of invalid or non-numeric entries in a few rows: {misclassified_numeric}"
|
| 33 |
-
)
|
| 34 |
-
|
| 35 |
-
# 3. Checking duplicate records
|
| 36 |
-
num_duplicates = df.duplicated().sum()
|
| 37 |
-
if num_duplicates > 0:
|
| 38 |
-
issues_found = True
|
| 39 |
-
st.warning(f"⚠️ Your dataset contains {num_duplicates} duplicated rows.")
|
| 40 |
-
|
| 41 |
-
# 4. Checking Constant Columns (No Variation)
|
| 42 |
-
constant_cols = [col for col in df.columns if df[col].nunique(dropna=False) <= 1]
|
| 43 |
-
if constant_cols:
|
| 44 |
-
issues_found = True
|
| 45 |
-
st.warning(f"⚠️ These columns contain only a single unique value and may be useless for analysis: {constant_cols}")
|
| 46 |
-
|
| 47 |
-
# 5. Suspiciously High Cardinality in Categorical Columns
|
| 48 |
-
high_card_cols = [col for col in df.select_dtypes(include='object') if df[col].nunique() > 100]
|
| 49 |
-
if high_card_cols:
|
| 50 |
-
issues_found = True
|
| 51 |
-
st.warning(f"⚠️ These object-type columns have unusually high unique values (possibly IDs or noisy data): {high_card_cols}")
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
# Final message
|
| 55 |
-
if not issues_found:
|
| 56 |
-
st.success("✅ No major issues detected. Dataset looks clean!")
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
# Function that will identify the task of the dataset
|
| 60 |
-
def task_type(df: pd.DataFrame, target_col: str) -> str:
|
| 61 |
-
"""
|
| 62 |
-
Determine the machine learning task type based on the target column.
|
| 63 |
-
|
| 64 |
-
Args:
|
| 65 |
-
df (pd.DataFrame): The input dataset.
|
| 66 |
-
target_col (str or None): The target column name, or None for unsupervised tasks.
|
| 67 |
-
|
| 68 |
-
Returns:
|
| 69 |
-
str: One of the following task types:
|
| 70 |
-
- "Classification" if the target is categorical.
|
| 71 |
-
- "Clustering" if the target is not provided.
|
| 72 |
-
- "Regression" if the target is numerical.
|
| 73 |
-
- "Unknown" if the type cannot be recognized.
|
| 74 |
-
"""
|
| 75 |
-
# Handle unsupervised case (no target)
|
| 76 |
-
if target_col == "No Target":
|
| 77 |
-
return "Clustering"
|
| 78 |
-
|
| 79 |
-
target_series = df[target_col]
|
| 80 |
-
dtype = target_series.dtype
|
| 81 |
-
n_unique = target_series.nunique()
|
| 82 |
-
|
| 83 |
-
# Check for classification
|
| 84 |
-
if dtype == 'object' or dtype == 'bool' or (dtype == 'category'):
|
| 85 |
-
return "Classification"
|
| 86 |
-
|
| 87 |
-
# Check for binary/multi-class classification represented as integers or floats
|
| 88 |
-
if dtype.kind in ['i', 'u', 'f']: # Integer types
|
| 89 |
-
if n_unique <= 10: # Arbitrary threshold for classification
|
| 90 |
-
return "Classification"
|
| 91 |
-
else:
|
| 92 |
-
return "Regression"
|
| 93 |
-
|
| 94 |
-
# Numeric types default to regression
|
| 95 |
-
if dtype.kind in ['i', 'u', 'f']:
|
| 96 |
-
return "Regression"
|
| 97 |
-
|
| 98 |
-
return "Unknown"
|
| 99 |
-
|
| 100 |
-
# Function that will identify if an object feature is truly categorical or not
|
| 101 |
-
def is_probably_categorical(series: pd.Series, threshold_unique: int = 50, threshold_ratio: float = 0.1) -> bool:
|
| 102 |
-
"""
|
| 103 |
-
Determines whether a given pandas Series is likely to be a categorical feature.
|
| 104 |
-
|
| 105 |
-
Args:
|
| 106 |
-
series (pd.Series): The input data column to analyze.
|
| 107 |
-
threshold_unique (int, optional (default=50)) : Maximum number of unique values for an object-type column to be considered categorical.
|
| 108 |
-
threshold_ratio (float, optional (default=0.1)) : Maximum ratio of unique values to total entries for object-type column to be treated as categorical.
|
| 109 |
-
|
| 110 |
-
Returns:
|
| 111 |
-
bool: True if the series is likely categorical, False otherwise.
|
| 112 |
-
"""
|
| 113 |
-
|
| 114 |
-
# Heuristic for object types (e.g., strings): avoid classifying high-cardinality fields as categorical
|
| 115 |
-
if series.dtype == 'object':
|
| 116 |
-
num_unique = series.nunique()
|
| 117 |
-
unique_ratio = num_unique / len(series)
|
| 118 |
-
|
| 119 |
-
if num_unique <= threshold_unique and unique_ratio <= threshold_ratio:
|
| 120 |
-
return True # categorical
|
| 121 |
-
else:
|
| 122 |
-
return False # high-cardinality non-categorical (like names)
|
| 123 |
-
|
| 124 |
-
# Explicit categorical or boolean data types are considered categorical
|
| 125 |
-
elif pd.api.types.is_categorical_dtype(series):
|
| 126 |
-
return True
|
| 127 |
-
elif pd.api.types.is_bool_dtype(series):
|
| 128 |
-
return True
|
| 129 |
-
|
| 130 |
-
return False
|
| 131 |
-
|
| 132 |
-
# Function that will identify if an numerical feature is discrete or not
|
| 133 |
-
def is_discrete(series: pd.Series, max_unique: int = 20) -> bool:
|
| 134 |
-
"""
|
| 135 |
-
Determine whether a numeric series should be considered discrete.
|
| 136 |
-
|
| 137 |
-
Args:
|
| 138 |
-
series (pd.Series): The input numeric data column to analyze.
|
| 139 |
-
max_unique (int, optional): Maximum number of unique values allowed
|
| 140 |
-
to treat a column as discrete. Default is 20.
|
| 141 |
-
|
| 142 |
-
Returns:
|
| 143 |
-
bool: True if the series is likely discrete, False otherwise.
|
| 144 |
-
"""
|
| 145 |
-
# Check if the series is of integer type
|
| 146 |
-
if pd.api.types.is_integer_dtype(series):
|
| 147 |
-
return series.nunique() <= max_unique
|
| 148 |
-
|
| 149 |
-
if pd.api.types.is_float_dtype(series):
|
| 150 |
-
# If all values are whole numbers AND unique count is low → treat as discrete
|
| 151 |
-
if series.dropna().apply(float.is_integer).all():
|
| 152 |
-
return series.nunique() <= max_unique
|
| 153 |
-
|
| 154 |
-
return False
|
| 155 |
-
|
| 156 |
-
# Function that will identify if an numerical feature is continuous or not
|
| 157 |
-
def is_continuous(series: pd.Series, max_unique: int = 20) -> bool:
|
| 158 |
-
"""
|
| 159 |
-
Determine whether a numeric series is continuous.
|
| 160 |
-
|
| 161 |
-
Args:
|
| 162 |
-
series (pd.Series): The input numeric data column to analyze.
|
| 163 |
-
max_unique (int, optional): Threshold for unique values. If a float-type column
|
| 164 |
-
contains only whole numbers and has fewer than this count, it is not considered continuous.
|
| 165 |
-
Default is 20.
|
| 166 |
-
|
| 167 |
-
Returns:
|
| 168 |
-
bool: True if the series is likely continuous, False otherwise.
|
| 169 |
-
"""
|
| 170 |
-
# Only float types are considered potentially continuous
|
| 171 |
-
if pd.api.types.is_float_dtype(series):
|
| 172 |
-
# If it's float but looks like discrete, then not continuous
|
| 173 |
-
all_whole_numbers = series.dropna().apply(float.is_integer).all()
|
| 174 |
-
if all_whole_numbers and series.nunique() <= max_unique:
|
| 175 |
-
return False
|
| 176 |
-
return True
|
| 177 |
-
return False
|
| 178 |
-
|
| 179 |
-
# Function that will identify if an feature is date-time format and then extract the time-based components
|
| 180 |
-
def parse_datetime_columns(df: pd.DataFrame) -> tuple[pd.DataFrame, list, list]:
|
| 181 |
-
"""
|
| 182 |
-
Detects and parses datetime columns in a DataFrame, and extracts useful
|
| 183 |
-
date and/or time components into new columns.
|
| 184 |
-
|
| 185 |
-
Args:
|
| 186 |
-
df (pd.DataFrame): Input dataset.
|
| 187 |
-
|
| 188 |
-
Returns:
|
| 189 |
-
tuple:
|
| 190 |
-
- pd.DataFrame: Updated DataFrame with extracted datetime components.
|
| 191 |
-
- list: List of original columns identified as datetime.
|
| 192 |
-
- list: List of newly extracted datetime-related feature names.
|
| 193 |
-
"""
|
| 194 |
-
datetime_cols = []
|
| 195 |
-
extracted_datetime = []
|
| 196 |
-
today = pd.Timestamp.today() # Just the date, no time
|
| 197 |
-
|
| 198 |
-
for col in df.columns:
|
| 199 |
-
if pd.api.types.is_datetime64_any_dtype(df[col]):
|
| 200 |
-
datetime_cols.append(col)
|
| 201 |
-
elif df[col].dtype == "object":
|
| 202 |
-
try:
|
| 203 |
-
converted = pd.to_datetime(df[col], errors="raise")
|
| 204 |
-
df[col] = converted
|
| 205 |
-
datetime_cols.append(col)
|
| 206 |
-
except Exception:
|
| 207 |
-
continue
|
| 208 |
-
|
| 209 |
-
for col in datetime_cols:
|
| 210 |
-
# Flags for what actually exists
|
| 211 |
-
has_date = True
|
| 212 |
-
has_time = True
|
| 213 |
-
|
| 214 |
-
# Check if all dates are "today" → probably not originally present
|
| 215 |
-
# if df[col].dt.normalize().nunique() == 1 and df[col].dt.normalize().iloc[0] == today:
|
| 216 |
-
if (df[col].dt.year == today.year).all() or (df[col].dt.month == today.month).all() or (df[col].dt.day == today.day).all():
|
| 217 |
-
has_date = False
|
| 218 |
-
|
| 219 |
-
# Check if all times are 00:00:00 → probably not originally present
|
| 220 |
-
if (df[col].dt.hour == 0).all() and (df[col].dt.minute == 0).all() and (df[col].dt.second == 0).all():
|
| 221 |
-
has_time = False
|
| 222 |
-
|
| 223 |
-
if has_date:
|
| 224 |
-
df[f"{col}_year"] = df[col].dt.year
|
| 225 |
-
df[f"{col}_month"] = df[col].dt.month
|
| 226 |
-
df[f"{col}_day"] = df[col].dt.day
|
| 227 |
-
df[f"{col}_weekday"] = df[col].dt.day_name()
|
| 228 |
-
extracted_datetime.extend([
|
| 229 |
-
f"{col}_year", f"{col}_month", f"{col}_day", f"{col}_weekday"
|
| 230 |
-
])
|
| 231 |
-
else:
|
| 232 |
-
df[f"{col}_year"] = np.nan
|
| 233 |
-
df[f"{col}_month"] = np.nan
|
| 234 |
-
df[f"{col}_day"] = np.nan
|
| 235 |
-
df[f"{col}_weekday"] = np.nan
|
| 236 |
-
|
| 237 |
-
if has_time:
|
| 238 |
-
df[f"{col}_hour"] = df[col].dt.hour
|
| 239 |
-
df[f"{col}_minute"] = df[col].dt.minute
|
| 240 |
-
extracted_datetime.extend([
|
| 241 |
-
f"{col}_hour", f"{col}_minute"
|
| 242 |
-
])
|
| 243 |
-
else:
|
| 244 |
-
df[f"{col}_hour"] = np.nan
|
| 245 |
-
df[f"{col}_minute"] = np.nan
|
| 246 |
-
|
| 247 |
-
# Remove any columns that are now entirely NaN
|
| 248 |
-
df = df.dropna(axis=1, how='all')
|
| 249 |
-
|
| 250 |
-
return df, datetime_cols, extracted_datetime
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
src/app.py
DELETED
|
@@ -1,221 +0,0 @@
|
|
| 1 |
-
import streamlit as st
|
| 2 |
-
import numpy as np
|
| 3 |
-
import pandas as pd
|
| 4 |
-
from AutoVisualizer.processing import check_dataset_cleanliness, task_type,is_probably_categorical, is_discrete, is_continuous, parse_datetime_columns
|
| 5 |
-
from AutoVisualizer.categorical_viz import combine_figures_as_subplots, generate_count_plots, generate_bar_plots, generate_grouped_bar_plots, generate_pie_plots, generate_categorical_correlation_heatmap
|
| 6 |
-
from AutoVisualizer.numerical_viz import generate_box_plots, generate_numeric_correlation_heatmap, generate_scatter_plots, generate_histograms, generate_line_plots
|
| 7 |
-
|
| 8 |
-
# st.set_page_config(page_title= "Auto-Visualizer",page_icon= "📊",layout="wide")
|
| 9 |
-
|
| 10 |
-
with st.sidebar:
|
| 11 |
-
# Upload the dataset file (can upload only CSV, XLSX, JSON, XML)
|
| 12 |
-
uploaded_file = st.file_uploader("Upload your dataset file:", ["csv", "xlsx", "json", "xml"])
|
| 13 |
-
|
| 14 |
-
if uploaded_file is not None:
|
| 15 |
-
file_type = uploaded_file.name
|
| 16 |
-
|
| 17 |
-
try:
|
| 18 |
-
# Read the dataset through pandas
|
| 19 |
-
if file_type.endswith(".csv"):
|
| 20 |
-
df = pd.read_csv(uploaded_file)
|
| 21 |
-
elif file_type.endswith(".xlsx"):
|
| 22 |
-
df = pd.read_excel(uploaded_file)
|
| 23 |
-
elif file_type.endswith(".json"):
|
| 24 |
-
df = pd.read_json(uploaded_file)
|
| 25 |
-
else:
|
| 26 |
-
df = pd.read_xml(uploaded_file)
|
| 27 |
-
except Exception as e:
|
| 28 |
-
st.write("Error:", e)
|
| 29 |
-
|
| 30 |
-
with st.sidebar:
|
| 31 |
-
# Show a a Disclaimer to user to upload as clean data as possible
|
| 32 |
-
st.info("""
|
| 33 |
-
⚠️ **Heads up!** For the best experience, please upload a clean dataset.
|
| 34 |
-
|
| 35 |
-
This app is designed for *visualizing data*, not cleaning it.
|
| 36 |
-
|
| 37 |
-
📌 *Tip:* Use the quick checker below to spot potential issues.
|
| 38 |
-
""")
|
| 39 |
-
|
| 40 |
-
if st.button("Run Cleanliness Check"):
|
| 41 |
-
# Use session state to track button click
|
| 42 |
-
st.session_state.run_clean_check = True
|
| 43 |
-
st.divider()
|
| 44 |
-
|
| 45 |
-
# Display the cleanliness checker result on the main page (not sidebar) if button was clicked
|
| 46 |
-
if st.session_state.get("run_clean_check", False):
|
| 47 |
-
with st.expander("➡️ See Cleanliness Checker Result"):
|
| 48 |
-
check_dataset_cleanliness(df)
|
| 49 |
-
|
| 50 |
-
# Display the user DataFrame
|
| 51 |
-
st.markdown("Your Dataset:")
|
| 52 |
-
st.dataframe(df, height=210)
|
| 53 |
-
st.divider()
|
| 54 |
-
|
| 55 |
-
# Extract column names from the dataset + add "No Target" element if there's no target in the user dataset
|
| 56 |
-
feature_list = list(df.columns)
|
| 57 |
-
target_selector = ["No Target"] + feature_list
|
| 58 |
-
|
| 59 |
-
with st.sidebar:
|
| 60 |
-
# Ask the user to select target column from their dataset
|
| 61 |
-
target_col = st.selectbox("Specify the target column in your dataset:", target_selector)
|
| 62 |
-
# Identify the task/type of dataset (i.e. classification/regression/clustering(if no target feature at all))
|
| 63 |
-
task = task_type(df, target_col)
|
| 64 |
-
# Display the task to user
|
| 65 |
-
st.write(f"🔍 Task identified: **{task}**")
|
| 66 |
-
|
| 67 |
-
# Identify the date-time columns (if any) and extract new time-based components from it
|
| 68 |
-
# date_time_ls --> List that will store date-time feature names
|
| 69 |
-
# extracted_datetime --> List that will store extracted date-time feature names
|
| 70 |
-
df, date_time_ls, extracted_datetime = parse_datetime_columns(df)
|
| 71 |
-
|
| 72 |
-
# Remove date-time feature names as we already extracted time based components from it
|
| 73 |
-
feature_list = [x for x in feature_list if x not in date_time_ls]
|
| 74 |
-
|
| 75 |
-
categorical_ls = [] # List that will store categorical feature names
|
| 76 |
-
discrete_ls = [] # List that will store discrete feature names
|
| 77 |
-
continuous_ls = [] # List that will store continuous feature names
|
| 78 |
-
for feature in feature_list:
|
| 79 |
-
if is_probably_categorical(df[feature]):
|
| 80 |
-
categorical_ls.append(feature) # Calling Categorical Feature Identifier Function
|
| 81 |
-
elif is_discrete(df[feature]):
|
| 82 |
-
discrete_ls.append(feature) # Calling Discrete Feature Identifier Function
|
| 83 |
-
elif is_continuous(df[feature]):
|
| 84 |
-
continuous_ls.append(feature) # Calling Continuous Feature Identifier Function
|
| 85 |
-
# Add time based components that appear to be categorical
|
| 86 |
-
for feature in extracted_datetime:
|
| 87 |
-
if is_probably_categorical(df[feature]):
|
| 88 |
-
categorical_ls.append(feature)
|
| 89 |
-
|
| 90 |
-
# Creating Dialog Box to show identified features
|
| 91 |
-
@st.dialog("Identified/Extracted Features from your Dataset:-")
|
| 92 |
-
def open_dialog():
|
| 93 |
-
if categorical_ls:
|
| 94 |
-
with st.popover("Categorical Features", use_container_width= True):
|
| 95 |
-
st.code("\n".join([f"• {item}" for item in categorical_ls]))
|
| 96 |
-
if discrete_ls:
|
| 97 |
-
with st.popover("Discrete Features", use_container_width= True):
|
| 98 |
-
st.code("\n".join([f"• {item}" for item in discrete_ls]))
|
| 99 |
-
if continuous_ls:
|
| 100 |
-
with st.popover("Continuous Features", use_container_width= True):
|
| 101 |
-
st.code("\n".join([f"• {item}" for item in continuous_ls]))
|
| 102 |
-
if date_time_ls:
|
| 103 |
-
with st.popover("Date-Time Features", use_container_width= True):
|
| 104 |
-
st.code("\n".join([f"• {item}" for item in date_time_ls]))
|
| 105 |
-
with st.popover("Extracted features from your Date-Time like features", use_container_width= True):
|
| 106 |
-
st.code("\n".join([f"• {item}" for item in extracted_datetime]))
|
| 107 |
-
with st.sidebar:
|
| 108 |
-
# Calling the dialog box through a button
|
| 109 |
-
if st.button("See Your Feature Details"):
|
| 110 |
-
open_dialog()
|
| 111 |
-
|
| 112 |
-
# Generate the Plots
|
| 113 |
-
with st.spinner("Generating Plots.....", show_time= True):
|
| 114 |
-
if categorical_ls:
|
| 115 |
-
st.header("📊 Categorical Plots")
|
| 116 |
-
# 1. Count Plots
|
| 117 |
-
count_plots = []
|
| 118 |
-
for x_col in categorical_ls:
|
| 119 |
-
if df[x_col].nunique() <= 20:
|
| 120 |
-
count_plots.extend(generate_count_plots(df, x_col))
|
| 121 |
-
|
| 122 |
-
if count_plots:
|
| 123 |
-
st.subheader("Count Plots :-")
|
| 124 |
-
st.plotly_chart(combine_figures_as_subplots(count_plots), use_container_width=True)
|
| 125 |
-
|
| 126 |
-
# 2. Bar Plots
|
| 127 |
-
bar_plots = []
|
| 128 |
-
# (Categorical vs Discrete + Continuous)
|
| 129 |
-
for x_col in categorical_ls:
|
| 130 |
-
if df[x_col].nunique() <= 20:
|
| 131 |
-
bar_plots.extend(generate_bar_plots(df, x_col, discrete_ls + continuous_ls))
|
| 132 |
-
|
| 133 |
-
if bar_plots:
|
| 134 |
-
st.subheader("Bar Plots :-")
|
| 135 |
-
st.plotly_chart(combine_figures_as_subplots(bar_plots), use_container_width=True)
|
| 136 |
-
|
| 137 |
-
# 3. Grouped Bar Plots
|
| 138 |
-
grp_bar_plots = []
|
| 139 |
-
# (Categorical vs Discrete + Continuous)
|
| 140 |
-
grp_bar_plots.extend(generate_grouped_bar_plots(df, categorical_ls, discrete_ls + continuous_ls))
|
| 141 |
-
|
| 142 |
-
if grp_bar_plots:
|
| 143 |
-
st.subheader("Grouped Bar Plots :-")
|
| 144 |
-
st.plotly_chart(combine_figures_as_subplots(grp_bar_plots), use_container_width=True)
|
| 145 |
-
|
| 146 |
-
# 4. Pie Charts
|
| 147 |
-
pie_plots = []
|
| 148 |
-
for x_col in categorical_ls:
|
| 149 |
-
if df[x_col].nunique() <= 20:
|
| 150 |
-
pie_plots.extend(generate_pie_plots(df, x_col))
|
| 151 |
-
|
| 152 |
-
if pie_plots:
|
| 153 |
-
st.subheader("Pie Charts :-")
|
| 154 |
-
st.plotly_chart(combine_figures_as_subplots(pie_plots), use_container_width=True)
|
| 155 |
-
|
| 156 |
-
if continuous_ls:
|
| 157 |
-
st.header("📊 Numerical Plots")
|
| 158 |
-
# 5. Box Plots
|
| 159 |
-
box_plots = []
|
| 160 |
-
for x_col in categorical_ls:
|
| 161 |
-
if df[x_col].nunique() <= 10:
|
| 162 |
-
box_plots.extend(generate_box_plots(df, x_col, continuous_ls))
|
| 163 |
-
|
| 164 |
-
if box_plots:
|
| 165 |
-
st.subheader("Box Plots :-")
|
| 166 |
-
st.plotly_chart(combine_figures_as_subplots(box_plots), use_container_width=True)
|
| 167 |
-
|
| 168 |
-
# 6. Heat Maps
|
| 169 |
-
heat_maps = []
|
| 170 |
-
if task == 'Regression':
|
| 171 |
-
if categorical_ls:
|
| 172 |
-
heat_maps.extend(generate_categorical_correlation_heatmap(df, target_col, categorical_ls))
|
| 173 |
-
heat_maps.extend(generate_numeric_correlation_heatmap(df[continuous_ls]))
|
| 174 |
-
if heat_maps:
|
| 175 |
-
st.subheader("Heat Maps :-")
|
| 176 |
-
st.plotly_chart(combine_figures_as_subplots(heat_maps), use_container_width=True)
|
| 177 |
-
|
| 178 |
-
# 7. Scatter Plots
|
| 179 |
-
if len(continuous_ls) >= 2:
|
| 180 |
-
st.subheader("Scatter Plots")
|
| 181 |
-
scatter_plots = []
|
| 182 |
-
# Creation of unique feature pairs (no repetition like (B, A) if (A, B) is already used)
|
| 183 |
-
feature_pairs = []
|
| 184 |
-
for i in range(len(continuous_ls)):
|
| 185 |
-
for j in range(i + 1, len(continuous_ls)):
|
| 186 |
-
feature_pairs.append((continuous_ls[i], continuous_ls[j]))
|
| 187 |
-
selection = st.pills("Highlight using a categorical feature :- ", categorical_ls)
|
| 188 |
-
scatter_plots.extend(generate_scatter_plots(df, feature_pairs, selection))
|
| 189 |
-
if scatter_plots:
|
| 190 |
-
st.plotly_chart(combine_figures_as_subplots(scatter_plots), use_container_width=True)
|
| 191 |
-
|
| 192 |
-
# 8. Histograms
|
| 193 |
-
histograms = []
|
| 194 |
-
histograms.extend(generate_histograms(df, continuous_ls))
|
| 195 |
-
|
| 196 |
-
if histograms:
|
| 197 |
-
st.subheader("Histograms")
|
| 198 |
-
st.plotly_chart(combine_figures_as_subplots(histograms), use_container_width=True)
|
| 199 |
-
|
| 200 |
-
# 9. Line Plots
|
| 201 |
-
line_plots = []
|
| 202 |
-
if date_time_ls:
|
| 203 |
-
# Extract only date-related components
|
| 204 |
-
date_related_keywords = ['_year', '_month', '_day', '_weekday']
|
| 205 |
-
date_component_cols = [col for col in extracted_datetime if any(key in col for key in date_related_keywords)]
|
| 206 |
-
if date_component_cols:
|
| 207 |
-
st.subheader("Line Plots :-")
|
| 208 |
-
# Mapping of labels to values
|
| 209 |
-
time_grouping_options = {
|
| 210 |
-
"Daily": "D",
|
| 211 |
-
"Weekly": "W",
|
| 212 |
-
"Monthly": "ME",
|
| 213 |
-
"Yearly": "YE"
|
| 214 |
-
}
|
| 215 |
-
time_choice = st.pills("Choose time interval for grouping :- ", list(time_grouping_options.keys()))
|
| 216 |
-
# Extract the actual value for resampling
|
| 217 |
-
selected_freq = time_grouping_options[time_choice] if time_choice else "ME" # Use default "ME" if no selection
|
| 218 |
-
line_plots.extend(generate_line_plots(df, date_component_cols, continuous_ls, selected_freq))
|
| 219 |
-
|
| 220 |
-
if line_plots:
|
| 221 |
-
st.plotly_chart(combine_figures_as_subplots(line_plots), use_container_width= True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|