From de15a52dba43325c171584464995fa4311403e7e Mon Sep 17 00:00:00 2001 From: Primoz Date: Tue, 11 Oct 2022 08:36:23 +0000 Subject: [PATCH] Bug fix --- src/features/all_cleaning_individual/straw/main.py | 2 +- src/features/all_cleaning_overall/straw/main.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/features/all_cleaning_individual/straw/main.py b/src/features/all_cleaning_individual/straw/main.py index 8a222491..fdd3f496 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 b0baa760..40e346cc 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