Freature importantce graph fixed

Sohaib90 · Oct 7, 2020 · f478688 · f478688
1 parent dd6419f
commit f478688
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 0 deletions.
diff --git a/lib/__pycache__/dashboard.cpython-37.pyc b/lib/__pycache__/dashboard.cpython-37.pyc
diff --git a/lib/__pycache__/plotly_graphs.cpython-37.pyc b/lib/__pycache__/plotly_graphs.cpython-37.pyc
diff --git a/lib/plotly_graphs.py b/lib/plotly_graphs.py
@@ -142,6 +142,7 @@ def distributions(self, df, variable_name):
     def global_feature_impact_graph(self, df,classification=False):
 
         if classification==True:
+            df['class_name'] = df['class_name'].apply(str)
 
             fig = px.bar(df, x="VariableName", y="Impact_Value",
                          color='class_name', barmode='group',
@@ -173,6 +174,7 @@ def global_feature_impact_graph(self, df,classification=False):
     def global_feature_importance_graph(self, df, classification=False):
 
         if classification==True:
+            df['class_name'] = df['class_name'].apply(str)
 
             fig = px.bar(df, x="VariableName", y="Impact_Value",
                          color='class_name', barmode='group',