Skip to content

Commit f478688

Browse files
committed
Freature importantce graph fixed
1 parent dd6419f commit f478688

File tree

3 files changed

+2
-0
lines changed

3 files changed

+2
-0
lines changed
0 Bytes
Binary file not shown.
62 Bytes
Binary file not shown.

lib/plotly_graphs.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,7 @@ def distributions(self, df, variable_name):
142142
def global_feature_impact_graph(self, df,classification=False):
143143

144144
if classification==True:
145+
df['class_name'] = df['class_name'].apply(str)
145146

146147
fig = px.bar(df, x="VariableName", y="Impact_Value",
147148
color='class_name', barmode='group',
@@ -173,6 +174,7 @@ def global_feature_impact_graph(self, df,classification=False):
173174
def global_feature_importance_graph(self, df, classification=False):
174175

175176
if classification==True:
177+
df['class_name'] = df['class_name'].apply(str)
176178

177179
fig = px.bar(df, x="VariableName", y="Impact_Value",
178180
color='class_name', barmode='group',

0 commit comments

Comments
 (0)