baohan
/
scikit-learn
mirror of https://github.com/scikit-learn/scikit-learn.git
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

MNT hasattr->getattr on coef_ (#11698)

This commit is contained in:
Adrin Jalali 2018-08-12 09:20:14 +02:00 committed by Joel Nothman
parent a955da19ac
commit dd700f44df
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
sklearn/feature_selection/from_model.py
View File

@ -16,12 +16,13 @@ def _get_feature_importances(estimator, norm_order=1):
"""Retrieve or aggregate feature importances from estimator"""
importances = getattr(estimator, "feature_importances_", None)
if importances is None and hasattr(estimator, "coef_"):
coef_ = getattr(estimator, "coef_", None)
if importances is None and coef_ is not None:
if estimator.coef_.ndim == 1:
importances = np.abs(estimator.coef_)
importances = np.abs(coef_)
else:
importances = np.linalg.norm(estimator.coef_, axis=0,
importances = np.linalg.norm(coef_, axis=0,
ord=norm_order)
elif importances is None: