def _sort_features()

in sagemaker_notebook_instance/package/package/visuals.py [0:0]

• 19 lines of code
• 6 McCabe index (conditional complexity)

    def _sort_features(self, sort_order):
        assert sort_order in set([
            'absolute_ascending',
            'absolute_descending',
            'ascending',
            'descending'
        ])
        # find sorted index according to sort_order
        if sort_order.startswith('absolute'):
            values = np.abs(self._shap_values)
        else:
            values = self._shap_values
        if sort_order.endswith('ascending'):
            sorted_idxs = values.argsort().tolist()
        else:
            sorted_idxs = (-1 * values).argsort().tolist()
        # update features with sorted index
        self._shap_values = self._shap_values[sorted_idxs]
        self._names = [self._names[idx] for idx in sorted_idxs]
        if self._descriptions:
            self._descriptions = [self._descriptions[idx] for idx in sorted_idxs]