From 059c4a3515fdbd9873af28bc877e55f32b0c1486 Mon Sep 17 00:00:00 2001 From: Sarthak Pati Date: Wed, 19 Jul 2023 09:48:55 -0400 Subject: [PATCH 1/2] fixing import --- gandlf_collectStats | 7 ------- 1 file changed, 7 deletions(-) diff --git a/gandlf_collectStats b/gandlf_collectStats index 79fc45e8f..71a5955f4 100644 --- a/gandlf_collectStats +++ b/gandlf_collectStats @@ -7,15 +7,8 @@ import pandas as pd import seaborn as sns import matplotlib.pyplot as plt from pathlib import Path -from io import StringIO from GANDLF.cli import copyrightMessage -from GANDLF.utils.plot_utils import plot_all - -import os -import seaborn as sns -import matplotlib.pyplot as plt -from pathlib import Path def plot_all(df_training, df_validation, df_testing, output_plot_dir): From fc36c7d159fbc2802f117760a7283d2580700921 Mon Sep 17 00:00:00 2001 From: Sarthak Pati Date: Wed, 19 Jul 2023 09:50:10 -0400 Subject: [PATCH 2/2] added comment --- gandlf_collectStats | 1 + 1 file changed, 1 insertion(+) diff --git a/gandlf_collectStats b/gandlf_collectStats index 71a5955f4..1a79fa127 100644 --- a/gandlf_collectStats +++ b/gandlf_collectStats @@ -14,6 +14,7 @@ from GANDLF.cli import copyrightMessage def plot_all(df_training, df_validation, df_testing, output_plot_dir): """ Plots training, validation, and testing data for loss and other metrics. + TODO: this function needs to be moved under utils and then called after every training epoch. Args: df_training (pd.DataFrame): DataFrame containing training data.