Update report.py

danielmlow · web-flow · commit 7e276ff8279b · 2020-07-22T20:17:15.000-03:00
Made heatmap square
diff --git a/pydra_ml/report.py b/pydra_ml/report.py
@@ -35,7 +35,7 @@ def plot_summary(summary, output_dir=None, filename="shap_plot", plot_top_n_shap
         filename += f"_top_{plot_top_n_shap}"
 
     hm = sns.heatmap(
-        summary.round(3), annot=True, xticklabels=True, yticklabels=True, cbar=False
+        summary.round(3), annot=True, xticklabels=True, yticklabels=True, cbar=False, square=True,
     )
     hm.set_xticklabels(summary.columns, rotation=45)
     hm.set_yticklabels(summary.index, rotation=0)

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@ def plot_summary(summary, output_dir=None, filename="shap_plot", plot_top_n_shap`
`35`	`35`	`filename += f"_top_{plot_top_n_shap}"`
`36`	`36`
`37`	`37`	`hm = sns.heatmap(`
`38`		`- summary.round(3), annot=True, xticklabels=True, yticklabels=True, cbar=False`
	`38`	`+ summary.round(3), annot=True, xticklabels=True, yticklabels=True, cbar=False, square=True,`
`39`	`39`	`)`
`40`	`40`	`hm.set_xticklabels(summary.columns, rotation=45)`
`41`	`41`	`hm.set_yticklabels(summary.index, rotation=0)`