Skip to content

Commit

Permalink
Merge pull request #694 from mlcommons/collectStats_fix
Browse files Browse the repository at this point in the history
Fixed import for collect stats
  • Loading branch information
sarthakpati authored Jul 19, 2023
2 parents 8a8e437 + fc36c7d commit bca8bbd
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions gandlf_collectStats
Original file line number Diff line number Diff line change
Expand Up @@ -7,20 +7,14 @@ 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):
"""
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.
Expand Down

0 comments on commit bca8bbd

Please sign in to comment.