| 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) |