feat: updated some graphs, todo: legends are not yet complete

This commit is contained in:
Robin Meersman 2025-12-17 16:52:29 +01:00
parent 46872274f4
commit 08f6dc2927

View file

@ -15,11 +15,13 @@ if __name__ == "__main__":
# execution time # execution time
plt.figure() plt.figure()
model_df["original_file_size_mb"] = model_df["original_file_size"] / 1e6 model_df["original_file_size_mb"] = model_df["original_file_size"] / 1e6
model_df["compression_time_s"] = model_df["compression_time"] / 1e9
model_df["decompression_time_s"] = model_df["decompression_time"] / 1e9
# compression # compression
sns.lineplot( sns.lineplot(
data=model_df, data=model_df,
x="original_file_size_mb", x="original_file_size_mb",
y="compression_time", y="compression_time_s",
hue="context_length", hue="context_length",
palette="Set1", palette="Set1",
markers=True, markers=True,
@ -30,7 +32,7 @@ if __name__ == "__main__":
sns.lineplot( sns.lineplot(
data=model_df, data=model_df,
x="original_file_size_mb", x="original_file_size_mb",
y="decompression_time", y="decompression_time_s",
hue="context_length", hue="context_length",
palette="Set1", palette="Set1",
markers=True, markers=True,
@ -127,4 +129,4 @@ if __name__ == "__main__":
plt.ylabel("Filename") plt.ylabel("Filename")
plt.legend() plt.legend()
plt.tight_layout() plt.tight_layout()
plt.savefig(f"./graphs/{model_type}_loss.png") plt.savefig(f"./graphs/{model_type}_loss.png")