diff options
Diffstat (limited to 'tensorflow/python/estimator')
-rw-r--r-- | tensorflow/python/estimator/canned/boosted_trees.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/tensorflow/python/estimator/canned/boosted_trees.py b/tensorflow/python/estimator/canned/boosted_trees.py index 62757ef588..c59b59b653 100644 --- a/tensorflow/python/estimator/canned/boosted_trees.py +++ b/tensorflow/python/estimator/canned/boosted_trees.py @@ -956,9 +956,7 @@ def _compute_feature_importances_per_tree(tree, num_features): return importances -def _compute_feature_importances(tree_ensemble, - num_features, - normalize=True): +def _compute_feature_importances(tree_ensemble, num_features, normalize): """Compute the feature importances. The higher the value, the more important the feature. @@ -972,6 +970,9 @@ def _compute_feature_importances(tree_ensemble, sorted_feature_idx: A list of feature_id which is sorted by its feature importance. feature_importances: A list of corresponding feature importance. + + Raises: + AssertionError: Trees are all empty or root node only when normalizing. """ tree_importances = [_compute_feature_importances_per_tree(tree, num_features) for tree in tree_ensemble.trees] @@ -996,7 +997,7 @@ class _BoostedTrees(estimator.Estimator): self._sorted_feature_columns = sorted(feature_columns, key=lambda tc: tc.name) - def experimental_feature_importances(self, normalize=True): + def experimental_feature_importances(self, normalize=False): """Compute the feature importances. The higher the value, the more important the corresponding feature. @@ -1005,9 +1006,9 @@ class _BoostedTrees(estimator.Estimator): normalize: If True, normalize the feature importances. Returns: - sorted_feature_names: A list of feature name which is sorted + sorted_feature_names: 1-D array of feature name which is sorted by its feature importance. - feature_importances: A list of corresponding feature importance. + feature_importances: 1-D array of the corresponding feature importance. Raises: ValueError: Empty ensemble. |