Pytorch_CNN-RNN/utils/Graphs.py

import matplotlib.pyplot as plt
import pandas as pd

df = pd.read_csv("./cnn_net_data.csv")

# To Regraph ROC, run main.py without training
# It will run validation sett and create it based on latest trained model

# MAKES EPOCH VS AVG LOSS GRAPH
plt.plot(df['Epoch'], df['Avg_loss'], label="Loss on Training")
plt.xlabel('Epoch')
plt.ylabel('Average Loss')
plt.title('Loss vs Epoch On Training & Validation data')

# MAKES EPOCH VS VALIDATION LOSS GRAPH
plt.plot(df['Epoch'], df['Val_loss'], label="Loss on Validation")
plt.savefig('./avgloss_epoch_curve.png')
plt.legend()
plt.show()


# ACCURACY VS THRESHOLD GRAPH