INFRA: qqplot

Add density plots when plotting qqplots to compare two data.
tskit-dev · Feb 12, 2024 · b6d5c33 · b6d5c33
1 parent ab5d0b6
commit b6d5c33
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/verification.py b/verification.py
@@ -70,6 +70,18 @@ def _plot_qq_compare(self, data1, data1_name, data2, data2_name):
         pyplot.title(f"compare_{data1_name}_{data2_name}")
         pyplot.savefig(f, dpi=72)
         pyplot.close("all")
+
+        sns.kdeplot(data1, color="b", fill=True, legend=False)
+        pyplot.xlabel(data1_name)
+        f = self._build_filename(data1_name, "histogram")
+        pyplot.savefig(f, dpi=72)
+        pyplot.close("all")
+
+        sns.kdeplot(data2, color="b", fill=True, legend=False)
+        pyplot.xlabel(data2_name)
+        f = self._build_filename(data2_name, "histogram")
+        pyplot.savefig(f, dpi=72)
+        pyplot.close("all")
 
 
 def model_list(loc, scale):