Updated heatmaps

script/fasttext_005.py

@@ -141,29 +141,30 @@ def heatmap(df):
     d = df.lat_rom.mean()
     e = df.lat_dig.mean()
     f = df.rom_dig.mean()
+    means = [a, b, c, d, e, f]
     df_heat = pd.DataFrame(zip([10, a, b, c], [a, 10, d, e], [b, d, 10, f], [c, e, f, 10]))
     df_heat.index = ['las', 'lat', 'rom', 'dig']
     df_heat.columns = ['las', 'lat', 'rom', 'dig']
     df_heat.mask(df_heat == 10, inplace=True)
-    return sns.heatmap(df_heat, vmin=0.25, vmax=4.00, annot=True, cmap='Greys')
+    return sns.heatmap(df_heat, vmin=min(means), vmax=max(means), annot=True, cmap='Greys')
 
 # Create and save heatmaps
 
 # 1. Full
 pairwise_full_heatmap = heatmap(pairwise_full).get_figure()
-pairwise_full_heatmap.savefig('/home/mribary/Dropbox/pyDigest/images/pairwise_full_heatmap.png')
+#pairwise_full_heatmap.savefig('/home/mribary/Dropbox/pyDigest/images/pairwise_full_heatmap.png')
 
 # 2. Minimum
 pairwise_min_heatmap = heatmap(pairwise_min).get_figure()
-pairwise_min_heatmap.savefig('/home/mribary/Dropbox/pyDigest/images/pairwise_min_heatmap.png')
+#pairwise_min_heatmap.savefig('/home/mribary/Dropbox/pyDigest/images/pairwise_min_heatmap.png')
 
 # 3. Maximum
 pairwise_max_heatmap = heatmap(pairwise_max).get_figure()
-pairwise_max_heatmap.savefig('/home/mribary/Dropbox/pyDigest/images/pairwise_max_heatmap.png')
+#pairwise_max_heatmap.savefig('/home/mribary/Dropbox/pyDigest/images/pairwise_max_heatmap.png')
 
 # 4. Weighted
 pairwise_weighted_heatmap = heatmap(pairwise_weighted).get_figure()
-pairwise_weighted_heatmap.savefig('/home/mribary/Dropbox/pyDigest/images/pairwise_weighted_heatmap.png')
+#pairwise_weighted_heatmap.savefig('/home/mribary/Dropbox/pyDigest/images/pairwise_weighted_heatmap.png')
 
 #2 Sorted LatinISE-ROMTEXT dataframes