diff options
Diffstat (limited to 'lib/parameters.py')
| -rw-r--r-- | lib/parameters.py | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/lib/parameters.py b/lib/parameters.py index 352e7c7..0653100 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -712,17 +712,15 @@ class ModelAttribute: self.param_values, with_nan=False, categorical_to_scalar=True ) - param_to_fit_param = dict() + mutual_info_result = mutual_info_regression(fit_parameters, self.data) + + self.mutual_information_cache = dict() j = 0 - for i in range(len(self.param_names)): + for i, param_name in enumerate(self.param_names): if not ignore_index[i]: - param_to_fit_param[i] = j + self.mutual_information_cache[param_name] = mutual_info_result[j] j += 1 - self.mutual_information_cache = mutual_info_regression( - fit_parameters, self.data - ) - return self.mutual_information_cache @classmethod |