def save_curve()

in utils/utils.py [0:0]

• 23 lines of code
• 2 McCabe index (conditional complexity)

def save_curve(args, save_dir, training_losses, test_clean_losses, 
               overall_accs, many_accs, median_accs, low_accs, f1s):
    plt.figure()
    plt.plot(training_losses, 'b', label='training_losses')
    plt.plot(test_clean_losses, 'g', label='test_clean_losses')
    plt.grid()
    plt.legend()
    plt.savefig(osp.join(save_dir, 'losses.png'))
    plt.close()

    plt.plot(overall_accs, 'm', label='overall_accs')
    if args.imbalance_ratio < 1:
        plt.plot(many_accs, 'r', label='many_accs')
        plt.plot(median_accs, 'g', label='median_accs')
        plt.plot(low_accs, 'b', label='low_accs')
    plt.grid()
    plt.legend()
    plt.savefig(osp.join(save_dir, 'test_accs.png'))
    plt.close()

    plt.plot(f1s, 'm', label='f1s')
    plt.grid()
    plt.legend()
    plt.savefig(osp.join(save_dir, 'test_f1s.png'))
    plt.close()