@@ -234,16 +234,17 @@ def plot_coefs(coef_list, model_list, title_model, column_order):
234
234
:param column_order: List of column names to preserve coloring
235
235
from previous coefficient plots.
236
236
"""
237
-
237
+
238
238
for j , frame in enumerate (coef_list ):
239
-
239
+
240
240
auc_ = model_list [j ].auc (valid = True )
241
241
title_ = title_model + ' Model: {j}\n GBM AUC: {auc:.2f}' .format (j = str (j + 1 ), auc = auc_ )
242
242
fig , ax_ = plt .subplots (figsize = (10 , 8 ))
243
243
_ = frame [column_order ].plot (kind = 'barh' ,
244
244
ax = ax_ ,
245
245
title = title_ ,
246
- colormap = 'gnuplot' )
246
+ edgecolor = ['black' ]* len (frame .index ),
247
+ colormap = 'cool' )
247
248
248
249
249
250
def cv_model_rank (valid , seed_ , model_name_list , nfolds = 5 ):
@@ -499,7 +500,7 @@ def plot_pd_ice(x_name, par_dep_frame, ax=None):
499
500
500
501
# plot ICE
501
502
par_dep_frame .drop ('partial_dependence' , axis = 1 ).plot (x = x_name ,
502
- colormap = 'gnuplot ' ,
503
+ colormap = 'cool ' ,
503
504
ax = ax )
504
505
# overlay partial dependence, annotate plot
505
506
par_dep_frame .plot (title = 'Partial Dependence with ICE: ' + x_name ,
@@ -514,7 +515,7 @@ def plot_pd_ice(x_name, par_dep_frame, ax=None):
514
515
515
516
# plot ICE
516
517
par_dep_frame .drop ('partial_dependence' , axis = 1 ).plot (x = x_name ,
517
- colormap = 'gnuplot ' ,
518
+ colormap = 'cool ' ,
518
519
ax = ax )
519
520
520
521
# overlay partial dependence, annotate plot
0 commit comments