tdpeuter/2025ML-project-neural_compression
tdpeuter/2025ML-project-neural_compression
Archived
1
Fork 0
Code Issues Pull requests Releases 1 Activity Actions

fix: accuracy replaced by MSE loss, updated graphs

Browse source
This commit is contained in:
RobinMeersman 2025-12-16 18:12:10 +01:00
parent 5bf45e47a5
commit 9cd37f156a
15 changed files with 38 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

10
make_graphs.py
View file

@ -38,7 +38,7 @@ if __name__ == "__main__":
plt.tight_layout()
plt.savefig(f"./graphs/{model_type}_{dataset_type}_decompression_time.png")
# accuracy
# loss
plt.figure(figsize=(10, 4))
bar_height = 0.25
files = model_df["input_file_name"].unique()
@ -48,20 +48,20 @@ if __name__ == "__main__":
plt.barh(
y - bar_height / 2,
c256["match_percentage"] * 100,
c256["mse_loss"],
height=bar_height,
label="256"
)
plt.barh(
y + bar_height / 2,
c128["match_percentage"] * 100,
c128["mse_loss"],
height=bar_height,
label="128"
)
plt.yticks(y, files, rotation=45, ha="right")
plt.title(f"{model_type.capitalize()} accuracy for different context lengths")
plt.xlabel("Accuracy")
plt.title(f"{model_type.capitalize()} MSE loss for different context lengths")
plt.xlabel("MSE loss")
plt.ylabel("Filename")
plt.legend()
plt.tight_layout()