Untitled

 avatar
unknown
plain_text
5 months ago
1.2 kB
9
Indexable
import matplotlib.pyplot as plt
import numpy as np

plt.rcParams['font.size'] = 7.5# Set to your desired size

species = ("MMLU Pro", "BBH", "GPQA", "IfEval", "MUSR", "GSM8K")
penguin_means = {
    'BF16': (43.63,     59.02,  36.41,37.58,    43.68,  80.21   ),
    'FP8 + Quark': (43.68,      58.66,  36.25,  36.06,  43.55,  81.43),
    'FP8 + AutoFP8': (43.62,    58.64,  34.37,          34.97,  42.64   ,81.35),
}

colors_group = ['orange', 'green', 'deepskyblue']


x = np.arange(len(species))  # the label locations
width = 0.2  # the width of the bars
multiplier = 0

fig, ax = plt.subplots(layout='constrained')

i = 0
for attribute, measurement in penguin_means.items():
    offset = width * multiplier
    rects = ax.bar(x + offset, measurement, width, label=attribute, color=colors_group[i])
    ax.bar_label(rects, padding=3)
    multiplier += 1
    i+= 1

# Add some text for labels, title and custom x-axis tick labels, etc.
ax.set_ylabel('Score')
ax.set_xlabel('Benchmark')
ax.set_title('Benchmark scores of 100B Command R+ 0824')
ax.set_xticks(x + width, species)
ax.legend(loc='upper left', ncols=3)
#ax.set_ylim(0, 100)
plt.ylim(20 , 90)
#plt.autoscale()
plt.show()
Editor is loading...
Leave a Comment