From 0f21273508654133a51d1e20e485c3de33dc2779 Mon Sep 17 00:00:00 2001 From: Primoz Date: Wed, 12 Oct 2022 12:32:51 +0000 Subject: [PATCH] Bugs 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 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