Skip to content

Instantly share code, notes, and snippets.

@ubless607
ubless607 / plot_learning_curve.py
Last active March 14, 2022 10:08
A simple function for plotting a learning curve of the model
def plot_learning_curve(log_df,
metric_name='loss',
rolling=False,
window_size=5,
ylim=(None, None), **kwargs):
'''
A simple function for plotting a learning curve of the model
Args:
log_df: input pandas Dataframe