import os
import pandas as pd
import matplotlib.pyplot as plt

file_name = 'SPI&SAI.csv'
file = pd.read_csv(os.path.abspath(file_name))

# plot loss & spi & sai
plt.figure()
plt.subplot(311)
plt.plot(file['loss'], 'b')
plt.ylabel('Reconstruction Loss')
plt.subplot(312)
plt.plot(file['spi'], 'b')
plt.ylabel('SPI')
plt.subplot(313)
plt.plot(file['sai'], 'b')
plt.xlabel('Compression Ratio')
plt.ylabel('SAI')
# plt.savefig(file_name)
plt.show()