From f13388fca98908b70f4b6f4db8098924bf0e51fb Mon Sep 17 00:00:00 2001
From: Jim Schwoebel <30424731+jim-schwoebel@users.noreply.github.com>
Date: Fri, 14 Aug 2020 16:54:18 -0400
Subject: [PATCH] Update visualize.py

---
 visualize/visualize.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/visualize/visualize.py b/visualize/visualize.py
index 8d1e0478..8fcc7d1d 100644
--- a/visualize/visualize.py
+++ b/visualize/visualize.py
@@ -573,7 +573,7 @@ def visualize_features(classes, problem_type, curdir, default_features, balance_
 		fig=sns.violinplot(x=newdata['classes'], y=newdata[new_labels_[j]])
 		fig = fig.get_figure()
 		fig.tight_layout()
-		fig.savefig('%s_%s.png'%(str(j), feature_labels[0][j]))
+		fig.savefig('%s_%s.png'%(str(j), new_labels_[j]))
 		plt.close(fig)
 
 	os.mkdir('feature_plots_transformed')
@@ -586,7 +586,7 @@ def visualize_features(classes, problem_type, curdir, default_features, balance_
 		fig=sns.violinplot(x=newdata['classes'], y=newdata[new_labels[j]])
 		fig = fig.get_figure()
 		fig.tight_layout()
-		fig.savefig('%s_%s.png'%(str(j), feature_labels[0][j]))
+		fig.savefig('%s_%s.png'%(str(j), new_labels[j]))
 		plt.close(fig)
 		
 	##################################################