Update core/plot.py

core/plot.py

@@ -26,4 +26,17 @@ def plot_metrics(result):
     metrics = result['metrics']
     sns.barplot(x=list(metrics.keys()), y=list(metrics.values()), ax=ax)
     ax.set_title("Error Metrics")
+    return fig
+
+def plot_loss_curve(result):
+    train_losses = result['train_loss']
+    val_losses = result.get('val_loss', [])
+    fig, ax = plt.subplots()
+    ax.plot(train_losses, label='Train Loss')
+    if val_losses:
+        ax.plot(val_losses, label='Validation Loss')
+    ax.set_title('Loss Curve')
+    ax.set_xlabel('Epoch')
+    ax.set_ylabel('Loss')
+    ax.legend()
     return fig