Skip to content

Commit 117c566

Browse files
committed
remove legend frame, change box to bar
Signed-off-by: Sunyanan Choochotkaew <[email protected]>
1 parent fe75985 commit 117c566

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

cmd/cmd_plot.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ def metadata_plot(args, energy_source, metadata_df, output_folder, name):
134134
ax = axes
135135
else:
136136
ax = axes[i]
137-
sns.boxplot(data=metadata_df, x="feature_group", y="mae", hue="trainer", ax=ax, hue_order=sorted(metadata_df['trainer'].unique()), showfliers=False, palette="Set3")
137+
sns.barplot(data=metadata_df, x="feature_group", y="mae", hue="trainer", ax=ax, hue_order=sorted(metadata_df['trainer'].unique()), errorbar=None, palette="Set3")
138138
ax.set_title(component)
139139
ax.set_ylabel("MAE (Watt)")
140140
ax.set_xlabel("Feature Group")
@@ -144,6 +144,7 @@ def metadata_plot(args, energy_source, metadata_df, output_folder, name):
144144
# ax.legend(bbox_to_anchor=(1.05, 1.05))
145145
plt.suptitle("Pipieline metadata of {} {}".format(energy_source.upper(), args.output_type))
146146
plt.tight_layout()
147+
plt.legend(frameon=False)
147148
filename = os.path.join(output_folder, name + ".png")
148149
fig.savefig(filename)
149150
plt.close()

0 commit comments

Comments
 (0)