feat: updated some graphs, todo: legends are not yet complete
This commit is contained in:
parent
46872274f4
commit
08f6dc2927
1 changed files with 5 additions and 3 deletions
|
|
@ -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")
|
||||||
|
|
|
||||||
Reference in a new issue