This PR updates and adds some functionality of result.plot_corner().
It addresses the following problems:
Previously, the font size of the legend was fixed to 50 which was sufficient for a full 14 dimensional posterior plot. However, if only a few parameters are included via parameters = ["chirp_mass", "mass_ratio"], the legend was so large that it overlapped completely with the corner plot.
It was not possible to pass the truth values to the corner plot which is helpful for injections.
So far, the DINGO-IS samples were not plotted when weights was not included in the parameters list. If the user wanted to only use e.g. parameters = ["chirp_mass", "mass_ratio"], it would only plot the DINGO samples.
After this PR, the user can:
specify the font size of the legend via legend_font_size=... and adjust it for a lower dimensional corner plot.
visualize truth values via truths=... as well as the color truth_color=....
plot the DINGO-IS samples although weights are not specified as parameters.
Illustration of an example before the PR where the importance weighted samples are not shown due to the bug:
How the plot looks after the PR (when specifying an appropriate font size for the legend):
This PR updates and adds some functionality of
result.plot_corner()
.It addresses the following problems:
parameters = ["chirp_mass", "mass_ratio"]
, the legend was so large that it overlapped completely with the corner plot.DINGO-IS
samples were not plotted whenweights
was not included in the parameters list. If the user wanted to only use e.g.parameters = ["chirp_mass", "mass_ratio"]
, it would only plot theDINGO
samples.After this PR, the user can:
legend_font_size=...
and adjust it for a lower dimensional corner plot.truths=...
as well as the colortruth_color=...
.DINGO-IS
samples although weights are not specified as parameters.Illustration of an example before the PR where the importance weighted samples are not shown due to the bug:![image](https://github.com/dingo-gw/dingo/assets/116283499/84717249-52f0-4b46-bfbf-e2d20cb5f711)
How the plot looks after the PR (when specifying an appropriate font size for the legend):![image](https://github.com/dingo-gw/dingo/assets/116283499/eaf9337d-9d37-4f3b-a9cb-cc6efdeb161f)