Fixed hard coding of bins in get_univariate_plots

abhayspawar · Nov 8, 2018 · 4a10a66 · 4a10a66
1 parent c6cdc3e
commit 4a10a66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/featexp/base.py b/featexp/base.py
@@ -220,7 +220,7 @@ def get_univariate_plots(data, target_col, features_list=0, bins=10, data_test=0
         if cols != target_col and data[cols].dtype == 'O':
             print(cols + ' is categorical. Categorical features not supported yet.')
         elif cols != target_col and data[cols].dtype != 'O':
-            univariate_plotter(feature=cols, data=data, target_col=target_col, bins=10, data_test=data_test)
+            univariate_plotter(feature=cols, data=data, target_col=target_col, bins=bins, data_test=data_test)
 
 
 def get_trend_stats(data, target_col, features_list=0, bins=10, data_test=0):
@@ -263,4 +263,4 @@ def get_trend_stats(data, target_col, features_list=0, bins=10, data_test=0):
                                                                                    'Trend_correlation']
     if len(ignored) > 0:
         print('Categorical features ' + str(ignored) + ' ignored. Categorical features not supported yet.')
-    return (stats_all_df)
+    return (stats_all_df)