diff --git a/graph/graph.py b/graph/graph.py new file mode 100644 index 0000000..2b3f652 --- /dev/null +++ b/graph/graph.py @@ -0,0 +1,27 @@ +import matplotlib.pyplot as plt + + +models = ["mt5_small", "mt5_base", "mt5_large", "slovak-t5-base", "slovak_t5_base_small", "SlovakBERT"] +precision = [0.4923, 0.4533, 0.4949, 0.5000, 0.5385, 0.60] +recall = [0.6400, 0.6800, 0.9800, 0.6600, 0.9800, 0.7200] +f1_score = [0.5565, 0.5440, 0.6577, 0.5690, 0.6950, 0.6545] + + +plt.figure(figsize=(10, 6)) + +plt.plot(models, precision, marker='o', label="Presnosť (Precision)", color="orange") +plt.plot(models, recall, marker='o', label="Citlivosť (Recall)", color="red") +plt.plot(models, f1_score, marker='o', label="F1-Score", color="purple") + + +plt.title("Porovnanie modelov pre rozpoznávanie nenávistnej reči", fontsize=14) +plt.xlabel("Modely", fontsize=12) +plt.ylabel("Hodnota metrík", fontsize=12) +plt.xticks(rotation=15) +plt.ylim(0.0, 1.0) +plt.grid(True, linestyle='--', alpha=0.7) +plt.legend(loc="best", fontsize=10) + + +plt.tight_layout() +plt.show()