Skip to content

Commit a3102be

Browse files
authored
Update old_Sklearn.py
1 parent 8a3ea2b commit a3102be

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

SklearnV1d2/old_Sklearn.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -88,10 +88,10 @@
8888
# from_estimator method is non existent in older versions, but what about categorical features
8989
#fig.savefig('./part_disp_old.png', dpi=200)
9090

91-
#fig, ax = plt.subplots(figsize=(12, 6))
92-
#ax.set_title('GradientBoostingRegressor')
93-
#GBR_disp = plot_partial_dependence(model, X_NotNan, ['age', 'fare', ('age', 'fare')], ax=ax)
94-
#fig.savefig('./part_disp_old_NotCat.png', dpi=200)
91+
fig, ax = plt.subplots(figsize=(12, 6))
92+
ax.set_title('GradientBoostingRegressor')
93+
GBR_disp = plot_partial_dependence(model, X_NotNan, ['age', 'fare', ('age', 'fare')], ax=ax)
94+
fig.savefig('./part_disp_old_NotCat.png', dpi=200)
9595

9696
#fig, ax = plt.subplots(figsize=(12, 6))
9797
#ax.set_title('GradientBoostingRegressor')

0 commit comments

Comments
 (0)