model

app.py

@@ -16,20 +16,19 @@ def data_description(action_type):
         data_encoded[col] = label_encoding.fit_transform(data_encoded[col])
     input_data = data_encoded.drop(['Attrition'], axis=1)
     target_data = data_encoded[['Attrition']]
+    col_values = list(input_data.columns.values)
 
     if action_type == "Input Data":
         return input_data.head()
     if action_type == "Target Data":
         return target_data.head()
     if action_type == "Feature Selection By Mutual Information":
-        col_values = list(input_data.columns.values)
         feature_scores = mutual_info_classif(input_data, target_data)
         data = [["Feature", "Mutual Information (0: independent, 1: dependent)"]]
         for score, fname in sorted(zip(feature_scores, col_values), reverse=True)[:10]:
             data.append([fname, score])
         return data
     if action_type == "Feature Selection By Chi Square":
-        col_values = list(input_data.columns.values)
         feature_scores = chi2(input_data, target_data)[0]
         data = [["Feature", "Mutual Information (0: independent, 1: dependent)"]]
         for score, fname in sorted(zip(feature_scores, col_values), reverse=True)[:10]: