diff --git a/src/features/all_cleaning_individual/straw/main.py b/src/features/all_cleaning_individual/straw/main.py index f132ee62..0edd06c6 100644 --- a/src/features/all_cleaning_individual/straw/main.py +++ b/src/features/all_cleaning_individual/straw/main.py @@ -85,7 +85,7 @@ def straw_cleaning(sensor_data_files, provider): impute_w_sn2 = [col for col in features.columns if "homelabel" in col] features[impute_w_sn2] = features[impute_w_sn2].fillna(1) # Special case of imputation - nominal/ordinal value - impute_w_sn3 = [col for col features.columns if "loglocationvariance" in col] + impute_w_sn3 = [col for col in features.columns if "loglocationvariance" in col] features[impute_w_sn2] = features[impute_w_sn2].fillna(-1000000) # Special case of imputation - nominal/ordinal value diff --git a/src/features/all_cleaning_overall/straw/main.py b/src/features/all_cleaning_overall/straw/main.py index 9bb7bb3b..3250f6d7 100644 --- a/src/features/all_cleaning_overall/straw/main.py +++ b/src/features/all_cleaning_overall/straw/main.py @@ -80,7 +80,7 @@ def straw_cleaning(sensor_data_files, provider, target): impute_w_sn2 = [col for col in features.columns if "homelabel" in col] features[impute_w_sn2] = features[impute_w_sn2].fillna(1) # Special case of imputation - nominal/ordinal value - impute_w_sn3 = [col for col features.columns if "loglocationvariance" in col] + impute_w_sn3 = [col for col in features.columns if "loglocationvariance" in col] features[impute_w_sn2] = features[impute_w_sn2].fillna(-1000000) # Special case of imputation - loglocation # Impute selected phone features with 0