visualize multicore

visualize_multicore.py

def visualize(data, save=False):
    """
    plot the results
    
    """

    y = data.copy()  # get y-values
    x = y.pop('N')  # get x-values

    plt.plot(x, y)
    plt.title("Average Model Training Time")
    plt.xlabel('Sample Size ($n$)')
    plt.ylabel('Time (ms)')
    plt.legend([f'n_jobs={n}' for n in range(1, len(data.columns))])
    
    if save: plt.savefig('spread_the_love_new.png', dpi=250)
    
    plt.show()