Uni/machine-learning
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

graphmaker

Browse Source
This commit is contained in:
Tim
2025-08-22 13:12:34 +02:00
parent cf92ef246a
commit 379fcbf881
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
compare_graph_maker.py
View File

@@ -2,9 +2,9 @@ import os
import matplotlib.pyplot as plt
datasets = {
"cleaned": "games_march2025_cleaned",
"cleaned_2k": "games_march2025_cleaned_2k",
"cleaned_10k": "games_march2025_cleaned_10k"
#"cleaned": "games_march2025_cleaned",
"cleaned_2k": "games_march2025_cleaned_2k_i3k",
"cleaned_10k": "games_march2025_cleaned_10k_i3k"
}
# def results
results = {}
@@ -28,14 +28,14 @@ x = range(len(models))
plt.figure(figsize=(12,6))
#plt.bar([i - 0.25 for i in x], [results["cleaned"][m] for m in models], width=0.25, label="cleaned")
plt.bar(x, [results["cleaned_2k"][m] for m in models], width=0.5)#, label="cleaned_2k")
#plt.bar([i + 0.25 for i in x], [results["cleaned_10k"][m] for m in models], width=0.25, label="cleaned_10k")
plt.bar(x, [results["cleaned_2k"][m] for m in models], width=0.25, label="2k Dataset")
plt.bar([i + 0.25 for i in x], [results["cleaned_10k"].get(m,0) for m in models], width=0.25, label="10k Dataset")
plt.xticks(x, models, rotation=90)
plt.ylim(0, 1) # min max
plt.ylabel("Weighted F1-Score")
plt.title("Model Performance across Datasets")
#plt.legend()
plt.legend()
plt.tight_layout()
plt.savefig('compare_graph_latest.png')
plt.show()