Skip to content
Snippets Groups Projects
Commit 9696fd17 authored by Christian's avatar Christian
Browse files

using consistent colors for bar plot and line plots of action distributions

parent 6764e824
Branches
No related tags found
No related merge requests found
[
{
"dir": "mwoz",
"legend": "mwoz"
},
{
"dir": "scratch",
"legend": "scratch"
},
{
"dir": "sgd",
"legend": "sgd"
"legend": "SGD"
},
{
"dir": "mwoz",
"legend": "1%MWOZ"
},
{
"dir": "sgd_mw",
"legend": "sgd_mw"
"legend": "SGD->1%MWOZ"
}
]
\ No newline at end of file
......@@ -122,6 +122,6 @@ def create_bar_plots(action_distributions, alg_maps, possible_actions, output_di
df = pd.DataFrame(df_list, columns=['Probabilities'] + [alg_map["legend"] for alg_map in alg_maps])
fig = df.plot(x='Probabilities', kind='bar', stacked=False, title='Final Action Distributions',
rot=0, grid=True).get_figure()
rot=0, grid=True, color=sns.color_palette("husl", len(alg_maps))).get_figure()
plt.yticks(np.arange(0, 1, 0.1))
fig.savefig(os.path.join(output_dir, "final_action_probabilities.pdf"))
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment