control image size for analyze result (#812)

weiguoz · web-flow · commit 4a0506e87b0c · 2019-09-11T14:59:42.000+08:00
diff --git a/sql/template_analyze.go b/sql/template_analyze.go
@@ -79,8 +79,8 @@ bst.load_model(fname=model_path)
 explainer = shap.TreeExplainer(bst)
 shap_values = explainer.shap_values(X)
 
-shap.summary_plot(shap_values, X, **summaryAttrs)
-plt.savefig('summary')
+shap.summary_plot(shap_values, X, show=False, **summaryAttrs)
+plt.savefig('summary', bbox_inches='tight')
 `
 
 var analyzeTemplate = template.Must(template.New("analyzeTemplate").Parse(analyzeTemplateText))

Original file line number	Diff line number	Diff line change
`@@ -79,8 +79,8 @@ bst.load_model(fname=model_path)`
`79`	`79`	`explainer = shap.TreeExplainer(bst)`
`80`	`80`	`shap_values = explainer.shap_values(X)`
`81`	`81`
`82`		`-shap.summary_plot(shap_values, X, **summaryAttrs)`
`83`		`-plt.savefig('summary')`
	`82`	`+shap.summary_plot(shap_values, X, show=False, **summaryAttrs)`
	`83`	`+plt.savefig('summary', bbox_inches='tight')`
`84`	`84`	`
`85`	`85`
`86`	`86`	`var analyzeTemplate = template.Must(template.New("analyzeTemplate").Parse(analyzeTemplateText))`