INNER CODE UNIT · Python

plot_model_performances

alteryx/open_source_demos · predict-daily-temperature/utils.py:11

def plot_model_performances(baseline_evalml_score, random_forest_evalml_score, baseline_score, featuretools_score):
    # data to plot
    n_groups = 2

    evalml_runs = (baseline_evalml_score, random_forest_evalml_score)
    manual_runs = (baseline_score, featuretools_score)

    # create plot
    fig, ax = plt.subplots()
    index = np.arange(n_groups)
    bar_width = 0.35
    opacity = 0.8

    plt.bar(index, evalml_runs, bar_width,
            alpha=opacity,
            color='#9300ff',
            label='EvalML Runs')

View source record →