Create visualizations.py

utils/visualizations.py

@@ -0,0 +1,19 @@
+import matplotlib.pyplot as plt
+import seaborn as sns
+
+def plot_correlation_heatmap(df):
+    corr = df.corr()
+    plt.figure(figsize=(10, 8))
+    sns.heatmap(corr, annot=True, cmap="coolwarm", fmt=".2f", cbar=True)
+    plt.show()
+
+def plot_model_performance(metrics_df):
+    metrics_df.set_index('Model', inplace=True)
+    ax = metrics_df.plot(kind='bar', figsize=(10, 6), colormap='coolwarm', rot=45)
+    plt.title("Model Performance Metrics")
+    plt.ylabel("Scores")
+    plt.xlabel("Models")
+    plt.show()
+
+def save_plot_as_png(fig, filename):
+    fig.savefig(filename, format="png")