import matplotlib.pyplot as plt
import numpy as np

# Dáta z tabuľky
models = ["mT5-small", "mT5-base", "mT5-large", "SlovakT5-base", "SlovakT5-small"]
precision = [0.4923, 0.4533, 0.6473, 0.6689, 0.6230]
recall =    [0.6400, 0.6800, 0.5903, 0.6082, 0.5748]
f1_score =  [0.5662, 0.5440, 0.6188, 0.6386, 0.5989]

# Nastavenia grafu
x = np.arange(len(models))
width = 0.25

# Vykreslenie
fig, ax = plt.subplots(figsize=(10, 6))
ax.bar(x - width, precision, width, label='Precision', color='red')
ax.bar(x, recall, width, label='Recall', color='blue')
ax.bar(x + width, f1_score, width, label='F1-score', color='green')

# Popisy a štýl
ax.set_ylabel('Skóre')
ax.set_title('Porovnanie výkonnosti modelov (few-shot learning)')
ax.set_xticks(x)
ax.set_xticklabels(models, rotation=0, ha='center')
ax.set_ylim(0, 1)
ax.legend()
plt.grid(axis='y', linestyle='--', alpha=0.7)
plt.tight_layout()

# Zobrazenie alebo uloženie
plt.show()