diff --git a/miceforest/imputation_kernel.py b/miceforest/imputation_kernel.py index 0ea7361..f59c631 100644 --- a/miceforest/imputation_kernel.py +++ b/miceforest/imputation_kernel.py @@ -1738,12 +1738,12 @@ def get_feature_importance( Passed to lgb.feature_importance() normalize: bool - Whether to normalize the values within + Whether to normalize the values within each modeled variable to sum to 1. Returns ------- - pandas DataFrame of importance values. Rows are imputed + pandas DataFrame of importance values. Rows are imputed variables, and columns are predictor variables. """