import pickle
import matplotlib.pyplot as plt
import argparse

def main(picklefile):
    with open(picklefile, 'rb') as f:
        all_metrics = pickle.load(f)
    
    plt.plot([i for i in range(len(all_metrics))], [a[1] for a in all_metrics])
    plt.grid(True)
    plt.ylim(0.0, 0.05)
    plt.xlabel('Эпоха')
    plt.ylabel('MAE$')
    plt.show()
    
    plt.plot([i for i in range(len(all_metrics))], [a[0] for a in all_metrics])
    plt.grid(True)
    plt.ylim(0.0, 1.0)
    plt.xlabel('Эпоха')
    plt.ylabel('$R^2$')
    plt.show()

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Process a pickle file')
    parser.add_argument('picklefile', type=str, help='Path to pickle file')
    args = parser.parse_args()
    
    main(args.picklefile)