From 55517eb737463e21c6f6ad18f4711d38bd02fec5 Mon Sep 17 00:00:00 2001 From: Primoz Date: Wed, 12 Oct 2022 12:23:11 +0000 Subject: [PATCH] Necessary commit before proceeding. --- src/features/all_cleaning_individual/straw/main.py | 2 +- src/features/all_cleaning_overall/straw/main.py | 3 ++- 2 files changed, 3 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..f132ee62 100644 --- a/src/features/all_cleaning_individual/straw/main.py +++ b/src/features/all_cleaning_individual/straw/main.py @@ -100,7 +100,7 @@ def straw_cleaning(sensor_data_files, provider): col.startswith('phone_screen_rapids_') or col.startswith('phone_wifi_visible')] - features[impute_zero] = impute(features[impute_zero], method="zero") + features[impute_zero] = features[impute_zero].fillna(0) ## (5) STANDARDIZATION if provider["STANDARDIZATION"]: diff --git a/src/features/all_cleaning_overall/straw/main.py b/src/features/all_cleaning_overall/straw/main.py index b0baa760..9bb7bb3b 100644 --- a/src/features/all_cleaning_overall/straw/main.py +++ b/src/features/all_cleaning_overall/straw/main.py @@ -93,7 +93,8 @@ def straw_cleaning(sensor_data_files, provider, target): col.startswith('phone_messages_rapids_') or col.startswith('phone_screen_rapids_') or col.startswith('phone_wifi_visible')] - features[impute_zero] = impute(features[impute_zero], method="zero") + + features[impute_zero] = features[impute_zero].fillna(0) graph_bf_af(features, "5zero_imp")